diff --git a/systems/flake.nix b/systems/flake.nix
index f88658f..be8814d 100644
--- a/systems/flake.nix
+++ b/systems/flake.nix
@@ -3,9 +3,13 @@
 
   inputs = {
     nixpkgs.url = "github:NixOS/nixpkgs/release-23.11";
+    disko = {
+       url = github:nix-community/disko;
+       inputs.nixpkgs.follows = "nixpkgs";
+    }
   };
 
-  outputs = { self, nixpkgs, ... }@inputs: {
+  outputs = { self, nixpkgs, disko, ... }@inputs: {
     nixosConfigurations = {
       "vm" = nixpkgs.lib.nixosSystem {
         system = "x86_64-linux";
@@ -13,6 +17,11 @@
           # Import the configuration.nix we used before, so that the old configuration file can still take effect. 
           # Note: /etc/nixos/configuration.nix itself is also a Nix Module, so you can import it directly here
 #          ./configuration.nix
+           disko.nixosModules.disko
+           ./disko-config.nix
+           {
+              _module.args.disks = [ "/dev/vda" ];
+           }
           ./hardware-configuration.nix
 
           ({config, pkgs, ...}: {