diff --git a/module/execution/control.php b/module/execution/control.php
index fbb9c73d1f..5413b388d4 100644
--- a/module/execution/control.php
+++ b/module/execution/control.php
@@ -1097,6 +1097,7 @@ class execution extends control
         $isStage = isset($output['type']) && $output['type'] == 'stage';
         if(!empty($project) && in_array($project->model, $this->config->project->waterfallList))
         {
+            if($project->model == 'waterfall') $isStage = true;
             $this->view->parentStage  = isset($output['parentStage']) ? $output['parentStage'] : 0;
             $this->view->parentStages = $this->loadModel('programplan')->getParentStageList($projectID, 0, 0, 'withparent|noclosed|' . ($isStage ? 'stage' : 'notstage'));
         }