diff --git a/src/components/nodes/ModelParameters.tsx b/src/components/nodes/ModelParameters.tsx
index 61f4cd32..6ca6667c 100644
--- a/src/components/nodes/ModelParameters.tsx
+++ b/src/components/nodes/ModelParameters.tsx
@@ -211,6 +211,12 @@ function ModelParametersInner({
     };
   }, [useGrid]);
 
+  const displaySchema = useMemo(() => {
+    return useGrid && colCount > 1
+      ? reorderColumnFirst(sortedSchema, colCount)
+      : sortedSchema;
+  }, [sortedSchema, useGrid, colCount]);
+
   // Don't render anything for Gemini or if no model selected
   if (provider === "gemini" || !modelId) {
     return null;
@@ -221,12 +227,6 @@ function ModelParametersInner({
     return null;
   }
 
-  const displaySchema = useMemo(() => {
-    return useGrid && colCount > 1
-      ? reorderColumnFirst(sortedSchema, colCount)
-      : sortedSchema;
-  }, [sortedSchema, useGrid, colCount]);
-
   return (
     <div className="shrink-0">
       {error ? (