diff --git a/modules_forge/forge_canvas/canvas.py b/modules_forge/forge_canvas/canvas.py
index 39efd5c5..ff5fb5a3 100644
--- a/modules_forge/forge_canvas/canvas.py
+++ b/modules_forge/forge_canvas/canvas.py
@@ -10,9 +10,11 @@ create_or_modify_pyi_org = gradio.component_meta.create_or_modify_pyi
 
 
 def create_or_modify_pyi_org_patched(component_class, class_name, events):
-    if component_class.__name__ == 'LogicalImage':
-        return
-
+    try:
+        if component_class.__name__ == 'LogicalImage':
+            return
+    except:
+        pass
     return create_or_modify_pyi_org(component_class, class_name, events)