diff --git a/vscode/webviews/components/modelSelectField/ModelSelectField.story.tsx b/vscode/webviews/components/modelSelectField/ModelSelectField.story.tsx
index a19d6cdb6aa..554046b43d0 100644
--- a/vscode/webviews/components/modelSelectField/ModelSelectField.story.tsx
+++ b/vscode/webviews/components/modelSelectField/ModelSelectField.story.tsx
@@ -73,14 +73,6 @@ export const EnterpriseUser: Story = {
             isDotComUser: false,
             isCodyProUser: false,
         },
-    },
-}
-
-export const NewStyleEnterpriseUser: Story = {
-    args: {
-        userInfo: {
-            isDotComUser: false,
-            isCodyProUser: false,
-        },
+        serverSentModelsEnabled: true,
     },
 }
diff --git a/vscode/webviews/components/modelSelectField/ModelSelectField.tsx b/vscode/webviews/components/modelSelectField/ModelSelectField.tsx
index c9e9e78799a..771ea3ec09f 100644
--- a/vscode/webviews/components/modelSelectField/ModelSelectField.tsx
+++ b/vscode/webviews/components/modelSelectField/ModelSelectField.tsx
@@ -279,7 +279,7 @@ function modelAvailability(
     if (!userInfo.isDotComUser && !serverSentModelsEnabled) {
         return 'not-selectable-on-enterprise'
     }
-    if (isCodyProModel(model) && !userInfo.isCodyProUser) {
+    if (isCodyProModel(model) && userInfo.isDotComUser && !userInfo.isCodyProUser) {
         return 'needs-cody-pro'
     }
     return 'available'