diff --git a/test/test.py b/test/test.py
index 05001bdf..98f08275 100644
--- a/test/test.py
+++ b/test/test.py
@@ -46,8 +46,10 @@ def wait_for_window(pattern):
 
 def collect_state_ui(name):
     swaymsg(f"exec hakurei ps > '/tmp/{name}.ps'")
+    machine.wait_for_file(f"/tmp/{name}.ps")
     machine.copy_from_vm(f"/tmp/{name}.ps", "")
     swaymsg(f"exec hakurei --json ps > '/tmp/{name}.json'")
+    machine.wait_for_file(f"/tmp/{name}.json")
     machine.copy_from_vm(f"/tmp/{name}.json", "")
     machine.screenshot(name)