public function fixResponse(GetResponseForControllerResultEvent $event)
{
$data = $event->getControllerResult();
if ($data instanceof PluginResponseInterface) {
$response = $data;
} else {
$response = $this->pageResponseFactory->createPluginResponse($data);
}
$event->setResponse($response);
}