diff --git a/flake.nix b/flake.nix
index 2df0ace..13685a3 100644
--- a/flake.nix
+++ b/flake.nix
@@ -31,7 +31,7 @@
         };
 
         packages = pkgs.lib.lists.foldl (a: b: a // b) { } [
-          (pkgs.callPackage ./scripts/clean-s3-cache.nix { inherit inputs; })
+          (import ./scripts/clean-s3-cache.nix { inherit pkgs inputs; })
         ];
 
         hydraJobs = {
diff --git a/scripts/clean-s3-cache.nix b/scripts/clean-s3-cache.nix
index 77854ef..d952df2 100644
--- a/scripts/clean-s3-cache.nix
+++ b/scripts/clean-s3-cache.nix
@@ -1,8 +1,8 @@
-{ stdenvNoCC, python3, python3Packages, inputs }: rec {
-  clean-s3-cache-env = python3.buildEnv.override {
-    extraLibs = with python3Packages; [ boto3 ];
+{ pkgs, inputs }: rec {
+  clean-s3-cache-env = pkgs.python3.buildEnv.override {
+    extraLibs = with pkgs.python3Packages; [ boto3 ];
   };
-  clean-s3-cache = stdenvNoCC.mkDerivation {
+  clean-s3-cache = pkgs.stdenvNoCC.mkDerivation {
     pname = "clean-s3-cache";
     version = inputs.clean-s3-cache.lastModifiedDate;
     srcs = inputs.clean-s3-cache;