minor bug fixed in the UnitController.