Fix to pass example correctly