Update UnitManager.java