Fix small bug in update command