fix: updates for PR review

jaaydenh · jaaydenh · commit 440fedc2308e · 2025-04-16T16:31:50.000Z
diff --git a/site/src/pages/CreateWorkspacePage/CreateWorkspacePageExperimental.tsx b/site/src/pages/CreateWorkspacePage/CreateWorkspacePageExperimental.tsx
@@ -96,10 +96,6 @@ const CreateWorkspacePageExperimental: FC = () => {
 			return;
 		}
 
-		if (ws.current) {
-			ws.current.close();
-		}
-
 		const socket = API.templateVersionDynamicParameters(realizedVersionId, {
 			onMessage,
 			onError: (error) => {
@@ -110,9 +106,7 @@ const CreateWorkspacePageExperimental: FC = () => {
 		ws.current = socket;
 
 		return () => {
-			if (ws.current) {
-				ws.current.close();
-			}
+			socket.close();
 		};
 	}, [realizedVersionId, onMessage]);
 
