Small change to update_velocity methods