diff --git a/src/engines/BaseEngine.py b/src/engines/BaseEngine.py
index aade7ab..7624c99 100644
--- a/src/engines/BaseEngine.py
+++ b/src/engines/BaseEngine.py
@@ -1,5 +1,7 @@
-from abc import ABC, abstractmethod
 import gradio as gr
+from abc import ABC, abstractmethod
+
+from ..chore import GenerationContext
 
 
 class BaseEngine(ABC):
@@ -8,6 +10,7 @@ class BaseEngine(ABC):
     description: str
 
     def __init__(self):
+        self.ctx: GenerationContext  # This is for type hinting only
         pass
 
     @classmethod