diff --git a/packages/firmware.nix b/packages/firmware.nix
index b4771c5..923f09b 100644
--- a/packages/firmware.nix
+++ b/packages/firmware.nix
@@ -2,7 +2,7 @@
 let
   firmware = import inputs.glove80-zmk { inherit pkgs; };
 
-  keymap = "${../src}/main.dts";
+  keymap = "${../src}/main.keymap";
   kconfig = ../glove80.conf;
   extra_modules = [ inputs.zmk-helpers ];
 
diff --git a/packages/visual.nix b/packages/visual.nix
index ebe615d..cb6d29a 100644
--- a/packages/visual.nix
+++ b/packages/visual.nix
@@ -31,7 +31,7 @@ stdenvNoCC.mkDerivation {
 
   buildPhase = ''
     mkdir -p $out
-    ${keymapDrawer} -c ${configuration} parse -c 10 -z glove80.keymap > $out/layout.yaml
+    ${keymapDrawer} -c ${configuration} parse -c 10 -z ./src/main.keymap > $out/layout.yaml
     ${keymapDrawer} -c ${configuration} draw $out/layout.yaml > $out/layout.svg
   '';
 }
diff --git a/src/main.dts b/src/main.keymap
similarity index 100%
rename from src/main.dts
rename to src/main.keymap