diff --git a/overlays/nixpkgs/firefox.nix b/overlays/nixpkgs/firefox.nix
index 5c97d4d..7b78746 100644
--- a/overlays/nixpkgs/firefox.nix
+++ b/overlays/nixpkgs/firefox.nix
@@ -1,7 +1,12 @@
 # Shamelessly stolen most parts from https://github.com/schizofox/schizofox
 
-{ pkgs, ... }: final: prev: {
-  firefox-esr = pkgs.mkNixPak {
+{ pkgs, inputs, lib, ... }: final: prev: let
+  mkNixPak = inputs.nixpak.lib.nixpak {
+    inherit (pkgs) lib;
+    inherit pkgs;
+  };
+in {
+  firefox-esr = mkNixPak {
     config = { sloth, ... }: {
       app.package = pkgs.firefox-esr;
       flatpak.appId = "org.mozilla.firefox";