diff --git a/flake.nix b/flake.nix
index 729fce5..95d8573 100644
--- a/flake.nix
+++ b/flake.nix
@@ -23,12 +23,16 @@
 
         buildInputs = with pkgs; [ vulkan-headers vulkan-loader vulkan-validation-layers renderdoc ];
 
+        java = pkgs.jdk22;
+
         mkCustomShell = { extraPkgs ? [] }: pkgs.mkShell {
           LD_LIBRARY_PATH = pkgs.lib.makeLibraryPath buildInputs;
           VK_LAYER_PATH = "${pkgs.vulkan-validation-layers}/share/vulkan/explicit_layer.d:${pkgs.renderdoc}/share/vulkan/implicit_layer.d";
 
+          JAVA_HOME = java;
+
           packages = with pkgs; [
-            jdk22
+            java
             maven
           ] ++ extraPkgs;
         };