Fix typo after refactoring

2 files changed, 2 insertions, 2 deletions

diff --git a/app/Controller/BaseController.php b/app/Controller/BaseController.php
index 36871bb4..5f6976ad 100644
--- a/app/Controller/BaseController.php
+++ b/app/Controller/BaseController.php
@@ -77,7 +77,7 @@ abstract class BaseController extends Base
         $model = 'projectFile';
 
         if ($task_id > 0) {
-            $model = 'taskFile';
+            $model = 'taskFileModel';
             $project_id = $this->taskFinderModel->getProjectId($task_id);
 
             if ($project_id !== $this->request->getIntegerParam('project_id')) {
diff --git a/app/Controller/FileViewerController.php b/app/Controller/FileViewerController.php
index 245845c7..518f5b0b 100644
--- a/app/Controller/FileViewerController.php
+++ b/app/Controller/FileViewerController.php
@@ -45,7 +45,7 @@ class FileViewerController extends BaseController
         $type = $this->helper->file->getPreviewType($file['name']);
         $params = array('file_id' => $file['id'], 'project_id' => $this->request->getIntegerParam('project_id'));
 
-        if ($file['model'] === 'taskFile') {
+        if ($file['model'] === 'taskFileModel') {
             $params['task_id'] = $file['task_id'];
         }