diff --git a/flake.nix b/flake.nix
index f6a491a..072e060 100644
--- a/flake.nix
+++ b/flake.nix
@@ -33,7 +33,7 @@
         (system: nixpkgsFor.${system}.nixpkgs-fmt);
 
       overlays.default = final: prev: {
-        devShell = final.callPackage nixos/devShell { };
+        devShell = final.callPackage nixos/devShell { inherit self; };
         whisper_api = final.callPackage nixos/pkgs/whisper_api { inherit self; };
         # Our code is not compatible with pydantic version 2 yet.
         python3 = prev.python3.override {
diff --git a/nixos/devShell/default.nix b/nixos/devShell/default.nix
index 7d18c73..7cced74 100644
--- a/nixos/devShell/default.nix
+++ b/nixos/devShell/default.nix
@@ -1,16 +1,10 @@
-{ pkgs, ... }:
+{ self, pkgs, ... }:
 let
-  python-with-packages = pkgs.python3.withPackages (p: with p;   [
-    fastapi
-    ffmpeg-python
-    openai-whisper
-    python-multipart
-    uvicorn
-  ] ++ [
+  python-with-packages = pkgs.python3.withPackages (p: with p; [
     # only needed for development
     autopep8
     httpx
-  ]);
+  ] ++ self.packages.${pkgs.system}.whisper_api.propagatedBuildInputs);
 in
 pkgs.mkShell {
   buildInputs = with pkgs;[