diff --git a/packages/@vue/cli-ui/package.json b/packages/@vue/cli-ui/package.json
index a547d2372d..bbeaf19180 100644
--- a/packages/@vue/cli-ui/package.json
+++ b/packages/@vue/cli-ui/package.json
@@ -7,9 +7,9 @@
     "lint": "vue-cli-service lint",
     "graphql-api": "cross-env VUE_CLI_DEBUG=true vue-cli-service graphql-api",
     "run-graphql-api": "vue-cli-service run-graphql-api",
-    "run-test-graphql-api": "cross-env NODE_ENV=test VUE_APP_GRAPHQL_PORT=4040 VUE_APP_GRAPHQL_WS=ws://localhost:4040/graphql yarn run graphql-api",
+    "run-test-graphql-api": "cross-env NODE_ENV=test VUE_APP_GRAPHQL_PORT=4040 VUE_APP_CLI_UI_URL=ws://localhost:4040/graphql yarn run graphql-api",
     "prepublishOnly": "yarn run lint --no-fix && yarn run build",
-    "test:e2e:dev": "cross-env VUE_APP_GRAPHQL_WS=ws://localhost:4040/graphql vue-cli-service test:e2e --mode development",
+    "test:e2e:dev": "cross-env VUE_APP_CLI_UI_URL=ws://localhost:4040/graphql vue-cli-service test:e2e --mode development",
     "test:e2e:run": "vue-cli-service test:e2e --mode production --headless --url=http://localhost:4040",
     "test:e2e": "start-server-and-test run-test-graphql-api http://localhost:4040 test:e2e:dev",
     "test:run": "start-server-and-test run-test-graphql-api http://localhost:4040 test:e2e:run",