diff --git a/swagspace_cloud_init/__init__.py b/swagspace_cloud_init/__init__.py
index b316de4..adae01b 100644
--- a/swagspace_cloud_init/__init__.py
+++ b/swagspace_cloud_init/__init__.py
@@ -132,7 +132,7 @@ def build_vm():
 
     password = getpass(f"Password for user {args.username}:")
 
-    data = graphql_query(client, vm_query)["data"]
+    data = graphql_query(client, vm_query, variables={"name": args.vm_name})["data"]
     vm = data["vms"][0]
 
     cpu_cores = int(float(vm["vcpus"]))
diff --git a/swagspace_cloud_init/vm.graphql b/swagspace_cloud_init/vm.graphql
index b655be6..3d1692c 100644
--- a/swagspace_cloud_init/vm.graphql
+++ b/swagspace_cloud_init/vm.graphql
@@ -1,5 +1,5 @@
-query vms {
-  vms: virtual_machine_list(filters: {name: {exact: "test"}}) {
+query vms($name: String) {
+  vms: virtual_machine_list(filters: {name: {exact: $name}}) {
     name
     id
     vcpus