diff --git a/modules/scripts.py b/modules/scripts.py
index 4d0bbd665..1c964ac2c 100644
--- a/modules/scripts.py
+++ b/modules/scripts.py
@@ -525,8 +525,10 @@ def add_classes_to_gradio_component(comp):
     """
     this adds gradio-* to the component for css styling (ie gradio-button to gr.Button), as well as some others
     """
-
-    comp.elem_classes = ["gradio-" + comp.get_block_name(), *(comp.elem_classes or [])]
+    elem_classes = []
+    if hasattr(comp, "elem_classes"):
+        elem_classes = comp.elem_classes
+    comp.elem_classes = ["gradio-" + comp.get_block_name(), *(elem_classes)]
 
     if getattr(comp, 'multiselect', False):
         comp.elem_classes.append('multiselect')