diff --git a/module/upgrade/tao.php b/module/upgrade/tao.php
index eb7dcbc90c..1734f818cd 100644
--- a/module/upgrade/tao.php
+++ b/module/upgrade/tao.php
@@ -949,7 +949,7 @@ class upgradeTao extends upgradeModel
     protected function migrateClassifyToModule(object $group, int $groupID)
     {
         $this->app->loadLang('process');
-        $classifyKey  = $group->projectModel . 'Classify';
+        $classifyKey  = $group->projectModel == 'waterfall' ? 'classify' : $group->projectModel . 'Classify';
         $classifyList = $this->dao->select('`key`,`value`')->from(TABLE_LANG)->where('module')->eq('process')->andWhere('section')->eq($classifyKey)->fetchPairs();
         if(empty($classifyList)) $classifyList = $this->lang->process->classifyList;