diff --git a/Project.toml b/Project.toml
index e8785f590f..ac45ce7756 100644
--- a/Project.toml
+++ b/Project.toml
@@ -29,6 +29,7 @@ Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
+DataAugmentation = "0.1"
 Zygote = "0.6"
 julia = "1"