updated UnitsController specs | fixing