Fix a bug preventing in-air control and swimming.