diff --git a/src/Module/Templates/DemoWeb/src/Controllers/CommentController.php.tpl b/src/Module/Templates/DemoWeb/src/Controllers/CommentController.php.tpl
index ecdc0f0f..9d17687c 100644
--- a/src/Module/Templates/DemoWeb/src/Controllers/CommentController.php.tpl
+++ b/src/Module/Templates/DemoWeb/src/Controllers/CommentController.php.tpl
@@ -16,6 +16,7 @@ namespace {{MODULE_NAMESPACE}}\Controllers;
 
 use {{MODULE_NAMESPACE}}\Services\CommentService;
 use {{MODULE_NAMESPACE}}\DTOs\CommentDTO;
+use Quantum\Http\Response;
 use Quantum\Http\Request;
 
 /**
diff --git a/src/Module/Templates/Toolkit/src/Controllers/EmailsController.php.tpl b/src/Module/Templates/Toolkit/src/Controllers/EmailsController.php.tpl
index 4b8013fe..05c1b543 100644
--- a/src/Module/Templates/Toolkit/src/Controllers/EmailsController.php.tpl
+++ b/src/Module/Templates/Toolkit/src/Controllers/EmailsController.php.tpl
@@ -64,6 +64,6 @@ class EmailsController extends BaseController
     {
         $this->emailService->deleteEmail($emailId);
 
-        redirect(base_url(true) . '/emails');
+        return redirect(base_url(true) . '/emails');
     }
 }