diff --git a/Project.toml b/Project.toml
index d523c4d..92703d8 100644
--- a/Project.toml
+++ b/Project.toml
@@ -31,6 +31,7 @@ Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
 cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
 
 [compat]
+NeuroTreeModels = "1"
 julia = "1"
 
 [extras]