@Override
  public void restoreStagedModel(PortletDataContext portletDataContext, T stagedModel)
      throws PortletDataException {

    StagedModelRepository<T> stagedModelRepository = getStagedModelRepository();

    if (stagedModelRepository == null) {
      super.restoreStagedModel(portletDataContext, stagedModel);

      return;
    }

    stagedModelRepository.restoreStagedModel(portletDataContext, stagedModel);
  }