diff --git a/cli-mock/src/main.rs b/cli-mock/src/main.rs
index a30efc4..879fb3b 100644
--- a/cli-mock/src/main.rs
+++ b/cli-mock/src/main.rs
@@ -138,9 +138,10 @@ async fn delete_vm(mut client: BrainCliServiceClient<Channel>, uuid: &str) -> Re
 async fn update_vm_request(
     mut client: BrainCliServiceClient<Channel>,
     node_pubkey: &str,
+    uuid: &str, 
 ) -> Result<()> {
     let req = UpdateVmRequest {
-        uuid: String::new(),
+        uuid: uuid.to_string(),
         node_pubkey: node_pubkey.to_string(),
         vcpus: 4,
         memory_mb: 4096,
@@ -187,7 +188,7 @@ async fn main() -> Result<()> {
 
     let contracts = list_contracts(client.clone()).await?;
     for contract in contracts {
-        if let Err(e) = update_vm_request(client.clone(), &contract.uuid).await {
+        if let Err(e) = update_vm_request(client.clone(), &use_default_string(), &contract.uuid).await {
             log::error!("Received error when updating VM {}: {e:?}", &contract.uuid);
         }
     }