diff --git a/.gitignore b/.gitignore
index b572a8f4..be098396 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,6 +3,7 @@ __pycache__
 *.safetensors
 *.pth
 *.dev.js
+*.pyi
 .DS_Store
 /output/
 /ESRGAN/*
diff --git a/modules_forge/forge_canvas/canvas.py b/modules_forge/forge_canvas/canvas.py
index ce29759c..d7161874 100644
--- a/modules_forge/forge_canvas/canvas.py
+++ b/modules_forge/forge_canvas/canvas.py
@@ -14,7 +14,13 @@ from PIL import Image
 from io import BytesIO
 from gradio.context import Context
 from functools import wraps
-from modules.ui_components import FormComponent
+
+
+class FormComponent:
+    webui_do_not_create_gradio_pyi_thank_you = True
+
+    def get_expected_parent(self):
+        return gr.components.Form
 
 
 canvas_js_root_path = os.path.dirname(__file__)