public function setDataSource($dataSource)
    {
        if (!$dataSource instanceof DataSourceInterface) {
            $dataSource = $this->getServiceLocator()->get('DataGrid\Factory\DataSource')->get($dataSource);
        }
        $this->dataSource = $dataSource;
        return $this;
    }