diff --git a/shell.nix b/shell.nix
index 6057869..ef1b156 100644
--- a/shell.nix
+++ b/shell.nix
@@ -21,16 +21,10 @@ let
     installPhase = ''
       mkdir -pv $out/share/java $out/bin
       cp -r ${src} $out/share/java/${name}
-      makeWrapper ${pkgs.jdk}/bin/java $out/bin/CuReSim --add-flags "-jar $out/share/java/${name}/${name}.jar"
+      makeWrapper ${jre}/bin/java $out/bin/CuReSim --add-flags "-jar $out/share/java/${name}/${name}.jar"
     '';
   };
 in mkShell {
-  buildInputs = [
-    R
-    rPackages.immuneSIM
-    rPackages.Biostrings
-    rPackages.stringr
-    jdk
-    CuReSim
-  ];
+  buildInputs =
+    [ R rPackages.immuneSIM rPackages.Biostrings rPackages.stringr CuReSim ];
 }