UnitOptimizer: Bugfix