/**
  * @copydoc GridHandler::loadData()
  */
 protected function loadData($request, $filter)
 {
     // Let parent class get data from data provider.
     $workflowStages = parent::loadData($request, $filter);
     // Filter the data.
     if ($filter['allStages']) {
         return array_combine($workflowStages, $workflowStages);
     } else {
         return array($this->getStageId() => $this->getStageId());
     }
 }