diff --git a/test/services/immich.nix b/test/services/immich.nix
index 5ca967e..8a3dc38 100644
--- a/test/services/immich.nix
+++ b/test/services/immich.nix
@@ -136,7 +136,7 @@ let
     };
 in
 {
-  basic = pkgs.nixosTest {
+  basic = shb.test.runNixOSTest {
     name = "immich-basic";
 
     nodes.server = basic;
@@ -145,7 +145,7 @@ in
     testScript = commonTestScript;
   };
 
-  https = pkgs.nixosTest {
+  https = shb.test.runNixOSTest {
     name = "immich-https";
 
     nodes.server = https;
@@ -154,7 +154,7 @@ in
     testScript = commonTestScript;
   };
 
-  backup = pkgs.nixosTest {
+  backup = shb.test.runNixOSTest {
     name = "immich-backup";
 
     nodes.server = backup;
diff --git a/test/services/paperless.nix b/test/services/paperless.nix
index b0096ca..46d9c26 100644
--- a/test/services/paperless.nix
+++ b/test/services/paperless.nix
@@ -133,7 +133,7 @@ let
     };
 in
 {
-  basic = pkgs.nixosTest {
+  basic = shb.test.runNixOSTest {
     name = "paperless-basic";
 
     nodes.server = basic;
@@ -142,7 +142,7 @@ in
     testScript = commonTestScript;
   };
 
-  https = pkgs.nixosTest {
+  https = shb.test.runNixOSTest {
     name = "paperless-https";
 
     nodes.server = https;
@@ -151,7 +151,7 @@ in
     testScript = commonTestScript;
   };
 
-  sso = pkgs.nixosTest {
+  sso = shb.test.runNixOSTest {
     name = "paperless-https";
 
     nodes.server = sso;
@@ -160,7 +160,7 @@ in
     testScript = commonTestScript;
   };
 
-  backup = pkgs.nixosTest {
+  backup = shb.test.runNixOSTest {
     name = "paperless-backup";
 
     nodes.server = backup;