diff --git a/src/app/units/states/tasks/viewer/viewer.coffee b/src/app/units/states/tasks/viewer/viewer.coffee
index 0dde935439..01cc1c88fb 100644
--- a/src/app/units/states/tasks/viewer/viewer.coffee
+++ b/src/app/units/states/tasks/viewer/viewer.coffee
@@ -19,5 +19,5 @@ angular.module('doubtfire.units.states.tasks.viewer', [
 )
 
 .controller('TaskViewerStateCtrl', ($scope) ->
-  $scope.taskDefs = $scope.unit.taskDefinitions
+
 )
diff --git a/src/app/units/states/tasks/viewer/viewer.tpl.html b/src/app/units/states/tasks/viewer/viewer.tpl.html
index 358d6f33a2..a7ae3ef831 100644
--- a/src/app/units/states/tasks/viewer/viewer.tpl.html
+++ b/src/app/units/states/tasks/viewer/viewer.tpl.html
@@ -1 +1 @@
-<f-tasks-viewer [task-defs]="taskDefs" [unit]="unit"></f-tasks-viewer>
+<f-tasks-viewer [task-defs]="unit.taskDefinitions" [unit]="unit"></f-tasks-viewer>