diff --git a/system/services/home-assistant/default.nix b/system/services/home-assistant/default.nix
index 836dfda8..54cc58a3 100644
--- a/system/services/home-assistant/default.nix
+++ b/system/services/home-assistant/default.nix
@@ -27,14 +27,15 @@
       "tplink_tapo"
     ];
     customComponents =
+      let
+        hassPython = config.services.home-assistant.package.python;
+      in
       (with pkgs.home-assistant-custom-components; [
         (dirigera_platform.override {
-          dirigera = (pkgs.python312Packages.dirigera.overrideAttrs (final: prev: { doCheck = false; }));
+          dirigera = (hassPython.pkgs.dirigera.overrideAttrs (final: prev: { doCheck = false; }));
         })
       ])
-      ++ (with pkgs; [
-        (home-assistant.python.pkgs.callPackage ./customComponents/tapo_control.nix { })
-      ]);
+      ++ [ (hassPython.pkgs.callPackage ./customComponents/tapo_control.nix { }) ];
     config = {
       # Includes dependencies for a basic setup
       # https://www.home-assistant.io/integrations/default_config/