diff --git a/ee/cli/plugins/stack_upgrade.py b/ee/cli/plugins/stack_upgrade.py
index 304cdf1d..e0f33826 100644
--- a/ee/cli/plugins/stack_upgrade.py
+++ b/ee/cli/plugins/stack_upgrade.py
@@ -6,6 +6,7 @@ from ee.core.aptget import EEAptGet
 from ee.core.apt_repo import EERepo
 from ee.core.services import EEService
 from ee.core.fileutils import EEFileUtils
+from ee.core.shellexec import EEShellExec
 import configparser
 import os