Refactoring: added controlled middleware and changed response class

1 files changed, 2 insertions, 2 deletions

diff --git a/app/Controller/TaskFile.php b/app/Controller/TaskFile.php
index 2b0152a7..0fcd2d69 100644
--- a/app/Controller/TaskFile.php
+++ b/app/Controller/TaskFile.php
@@ -8,7 +8,7 @@ namespace Kanboard\Controller;
  * @package  controller
  * @author   Frederic Guillot
  */
-class TaskFile extends Base
+class TaskFile extends BaseController
 {
     /**
      * Screenshot
@@ -24,7 +24,7 @@ class TaskFile extends Base
             return $this->response->redirect($this->helper->url->to('task', 'show', array('task_id' => $task['id'], 'project_id' => $task['project_id'])), true);
         }
 
-        $this->response->html($this->template->render('task_file/screenshot', array(
+        return $this->response->html($this->template->render('task_file/screenshot', array(
             'task' => $task,
         )));
     }