Update data source to return entity with results.