diff --git a/scripts/modelpreview.py b/scripts/modelpreview.py
index da43dc2..589792c 100644
--- a/scripts/modelpreview.py
+++ b/scripts/modelpreview.py
@@ -45,7 +45,7 @@ def import_lora_module_builtin():
 				spec.loader.exec_module(additional_networks_builtin)
 				return additional_networks_builtin
 		except:
-			return None
+			pass
 	return None
 
 
diff --git a/style.css b/style.css
index 2139ece..f3da70b 100644
--- a/style.css
+++ b/style.css
@@ -65,6 +65,10 @@ div.main > div.wrap > div.contain {
 	flex: 1;
 }
 
+#tab_modelpreview_xd_interface[style*="display: block"] {
+	display: grid !important;
+}
+
 #tab_modelpreview_xd_interface code {
 	overflow: auto;
 }
@@ -130,6 +134,10 @@ div.main > div.wrap > div.contain {
 	grid-template-columns: auto;
 }
 
+#tab_modelpreview_xd_interface[limit-height] .modelpreview_xd_html_row > .modelpreview_xd_flexcolumn_row:has(> .hidden) {
+	grid-template-rows: auto;
+}
+
 #tab_modelpreview_xd_interface .footer-tags {
 	font-size: 85%;
 	opacity: 0.85;