public function addUnit(Unit\Unit $unit) { if (!in_array('units', $this->_loaded)) { return $this->_loaders->get('units')->getByID($unit->id); } return parent::addUnit($unit); }