diff --git a/controllers/DefaultController.php b/controllers/DefaultController.php
index 50e2323..db72b90 100644
--- a/controllers/DefaultController.php
+++ b/controllers/DefaultController.php
@@ -55,7 +55,7 @@ public function actionRun($goal)
     public function actionGen($tpl, $file = null)
     {
         $gen = Yii::createObject([
-            'class'     => 'hidev\goals\Template',
+            'class'     => 'hidev\goals\TemplateGoal',
             'template'  => $tpl,
             'file'      => $file ?: (basename($tpl) . '.php'),
         ]);
diff --git a/goals/ComposerJsonGoal.php b/goals/ComposerJsonGoal.php
index efd4867..b572017 100644
--- a/goals/ComposerJsonGoal.php
+++ b/goals/ComposerJsonGoal.php
@@ -11,8 +11,6 @@
 
 namespace hidev\goals;
 
-use hiqdev\collection\Manager;
-
 /**
  * Goal for composer.json
  */