fix UnitStrategy bug