Fixing bug in example