Update for "flying" on UnitController.cs