diff --git a/packages/checks/default.nix b/packages/checks/default.nix
index ab66725..d47413c 100644
--- a/packages/checks/default.nix
+++ b/packages/checks/default.nix
@@ -8,7 +8,7 @@
       };
 
       patroni = pkgs.callPackage ./patroni.nix {
-        patroniModule = self.nixosModules.patroni;
+        inherit (self) nixosModules;
       };
       searxng = pkgs.callPackage ./searxng.nix {
         inherit (self'.packages) searxng;
diff --git a/packages/checks/patroni.nix b/packages/checks/patroni.nix
index 46e9b9d..00905aa 100644
--- a/packages/checks/patroni.nix
+++ b/packages/checks/patroni.nix
@@ -1,4 +1,4 @@
-{ nixosTest, patroniModule }:
+{ nixosTest, nixosModules }:
 
 # taken from https://github.com/phfroidmont/nixpkgs/blob/patroni-module/nixos/tests/patroni.nix
 nixosTest (
@@ -15,7 +15,8 @@ nixosTest (
       in
       {
         imports = [
-          patroniModule
+          nixosModules.patroni
+          nixosModules.systemd-extras
         ];
 
         networking.interfaces.eth1.ipv4.addresses = pkgs.lib.mkOverride 0 [