diff --git a/src/ZFTool/Controller/CreateController.php b/src/ZFTool/Controller/CreateController.php
index e3dcb46..e4b1f04 100644
--- a/src/ZFTool/Controller/CreateController.php
+++ b/src/ZFTool/Controller/CreateController.php
@@ -188,6 +188,9 @@ public function methodAction()
         $classReflection = $fileReflection->getClass($class);
 
         $classGenerator = Generator\ClassGenerator::fromReflection($classReflection);
+        $classGenerator->addUse('Zend\Mvc\Controller\AbstractActionController')
+                       ->addUse('Zend\View\Model\ViewModel')
+                       ->setExtendedClass('AbstractActionController');
 
         if ($classGenerator->hasMethod($action . 'Action')) {
             return $this->sendError(