diff --git a/nixos/tests/vscodium.nix b/nixos/tests/vscodium.nix
index 51c294d432e5..eee205e69607 100644
--- a/nixos/tests/vscodium.nix
+++ b/nixos/tests/vscodium.nix
@@ -51,6 +51,8 @@ let
         enableOCR = true;
 
         testScript = ''
+          machine = ${name}
+
           @polling_condition
           def codium_running():
               machine.succeed('pgrep -x codium')