diff --git a/nixos/tests/test_privatestorage.py b/nixos/tests/test_privatestorage.py
index e1f34fa4f1b0f603168fe825871d8cb81f52d8ce..911779498a8926545c6c0a52109a240f621d8ce5 100644
--- a/nixos/tests/test_privatestorage.py
+++ b/nixos/tests/test_privatestorage.py
@@ -100,7 +100,7 @@ def test(
 
     # Make sure the fake Stripe API server is ready for requests.
     try:
-        api_stripe_com.wait_for_unit("api.stripe.com")
+        api_stripe_com.wait_for_open_port(80)
     except:
         code, output = api_stripe_com.execute('journalctl -u api.stripe.com')
         api_stripe_com.log(output)