diff --git a/app/dts/behaviors.dtsi b/app/dts/behaviors.dtsi
index 8fa30b34..58ee8359 100644
--- a/app/dts/behaviors.dtsi
+++ b/app/dts/behaviors.dtsi
@@ -19,4 +19,3 @@
 #include <behaviors/key_repeat.dtsi>
 #include <behaviors/backlight.dtsi>
 #include <behaviors/macros.dtsi>
-#include <behaviors/custom_lock.dtsi>
diff --git a/app/dts/behaviors/custom_lock.dtsi b/app/dts/behaviors/custom_lock.dtsi
deleted file mode 100644
index aa791813..00000000
--- a/app/dts/behaviors/custom_lock.dtsi
+++ /dev/null
@@ -1,15 +0,0 @@
-/*
- * Copyright (c) 2022 The ZMK Contributors
- *
- * SPDX-License-Identifier: MIT
- */
-
-/ {
-	behaviors {
-		/omit-if-no-ref/ lock: behavior_custom_lock {
-			compatible = "zmk,behavior-custom-lock";
-			label = "CUSTOM LOCK";
-			#binding-cells = <0>;
-		};
-	};
-};
diff --git a/app/dts/bindings/behaviors/zmk,behavior-custom-lock.yaml b/app/dts/bindings/behaviors/zmk,behavior-custom-lock.yaml
index 7ea9c5dc..34b67e17 100644
--- a/app/dts/bindings/behaviors/zmk,behavior-custom-lock.yaml
+++ b/app/dts/bindings/behaviors/zmk,behavior-custom-lock.yaml
@@ -1,8 +1,11 @@
 # Copyright (c) 2022 The ZMK Contributors
 # SPDX-License-Identifier: MIT
 
-description: Custom Lock Behavior
+description: Custom lock behavior variable definition
 
 compatible: "zmk,behavior-custom-lock"
 
-include: zero_param.yaml
+child-binding:
+  description: Key definitions for lock variable
+
+  include: two_param.yaml