public function load()
    {
        $models = $this->loadRaw();

        foreach ($models as & $model) {
            $model = $model->getRepo()->getIdentityMap()->get($model);
        }

        return new RepoModels($this->getRepo(), $models);
    }