diff --git a/flake.nix b/flake.nix
index 723bd6d..4873b3f 100755
--- a/flake.nix
+++ b/flake.nix
@@ -241,6 +241,13 @@
       inputs.home-manager.follows = "steamdeck-home-manager";
     };
 
+    steamdeck-disko = {
+      # the fork is needed for partition attributes support
+      url = "github:nvmd/disko/gpt-attrs";
+      # url = "github:nix-community/disko";
+      inputs.nixpkgs.follows = "steamdeck-nixpkgs";
+    };
+
     #####################################################
     #                   MacBook                         #
     #####################################################
@@ -317,6 +324,7 @@
       steamdeck-nixos-hardware,
       steamdeck-sops-nix,
       steamdeck-steam-rom-manager,
+      steamdeck-disko,
 
       # MacBook
       nix-darwin,
@@ -513,7 +521,7 @@
             inherit inputs outputs;
           };
           modules = [
-            pi4-disko.nixosModules.disko
+            steamdeck-disko.nixosModules.disko
             ./share/disko/disko.nix
             steamdeck-impermanence.nixosModules.impermanence
             steamdeck-lanzaboote.nixosModules.lanzaboote
diff --git a/share/disko/disko.nix b/share/disko/disko.nix
index f11eeab..365062e 100644
--- a/share/disko/disko.nix
+++ b/share/disko/disko.nix
@@ -1,6 +1,6 @@
 { ... }:
 let
-  rootDisk = "/dev/sda";
+  rootDisk = "/dev/nvme0n1";
 in
 {
   disko.devices.disk.main.imageSize = "32G";