diff --git a/terraform/scripts/stage_talos_image.py b/terraform/scripts/stage_talos_image.py
index 6ad0880..926d042 100644
--- a/terraform/scripts/stage_talos_image.py
+++ b/terraform/scripts/stage_talos_image.py
@@ -36,8 +36,6 @@ def ensure_image(base_url, token, node, node_address, storage, image_url, filena
             return filename, False
 
     source_filename = filename
-    if network_config_path:
-        source_filename = f"base-{filename}"
 
     response = api_request(base_url, token, "GET", content_path)
     source_present = any(item.get("volid", "").endswith(source_filename) for item in response.get("data", []))
@@ -49,6 +47,7 @@ def ensure_image(base_url, token, node, node_address, storage, image_url, filena
             "url": image_url,
         })
         wait_for_task(base_url, token, node, task.get("data"))
+        wait_for_storage_content(base_url, token, node, storage, filename)
 
     if network_config_path:
         build_custom_iso(base_url, token, node_address, storage, source_filename, filename, network_config_path)