diff --git a/artiq/shell.nix b/artiq/shell.nix
index edf5ade..0f14cc3 100644
--- a/artiq/shell.nix
+++ b/artiq/shell.nix
@@ -4,5 +4,5 @@ let
   artiqpkgs = import ./default.nix { inherit pkgs; };
 in
   pkgs.mkShell {
-    buildInputs = with artiqpkgs; [ binutils-or1k llvm-or1k llvmlite-artiq artiq ];
+    buildInputs = [ (pkgs.python3.withPackages(ps: [artiqpkgs.artiq])) ];
   }