diff --git a/vscode/react/src/pages/lineage.tsx b/vscode/react/src/pages/lineage.tsx
index d2d4e85858..546e1ad4bf 100644
--- a/vscode/react/src/pages/lineage.tsx
+++ b/vscode/react/src/pages/lineage.tsx
@@ -110,6 +110,8 @@ function Lineage() {
       fetchFirstTimeModelIfNotSet(models).then(modelName => {
         if (modelName && selectedModel === undefined) {
           setSelectedModel(modelName)
+        } else {
+          setSelectedModel(models[0].name)
         }
       })
     }