Fix example broken by recent changes