diff --git a/flake.nix b/flake.nix
index 9744941..5e18c47 100644
--- a/flake.nix
+++ b/flake.nix
@@ -70,6 +70,10 @@ packages.x86_64-linux.ozai-webui-run = pkgs.stdenv.mkDerivation rec {
 
   installPhase = ''
     mkdir -p $out/bin
+    mkdir -p $out/share/ozai_webui/static
+    mkdir -p $out/share/ozai_webui/templates
+    mkdir -p $out/bin/templates
+    mkdir -p $out/bin/static
     cp -r $src/templates $out/bin/templates
     cp -r $src/static $out/bin/static
     cp $src/run.sh $out/bin/run