diff --git a/geoapps_utils/base.py b/geoapps_utils/base.py
index c0a402d8..2cb2325f 100644
--- a/geoapps_utils/base.py
+++ b/geoapps_utils/base.py
@@ -101,7 +101,7 @@ def read_ui_json(cls, filepath: str | Path, **kwargs) -> InputFile:
         return InputFile.read_ui_json(filepath, validations=cls._validations, **kwargs)
 
     @classmethod
-    def start(cls, filepath: str | Path | InputFile, mode="r+", **kwargs) -> Driver:
+    def start(cls, filepath: str | Path | InputFile, mode="r+", **kwargs) -> Self:
         """
         Run application specified by 'filepath' ui.json file.