Fix error in really simple code