diff --git a/test/Project.toml b/test/Project.toml
index 32174a7..44c3c2d 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -6,6 +6,7 @@ Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
 InferOpt = "4846b161-c94e-4150-8dac-c7ae193c601f"
 JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
 JuliaFormatter = "98e50ef6-434e-11e9-1051-2b60c6c9e899"
+LoweredCodeUtils = "6f1432cf-f94c-5a45-995e-cdbf5db27b0b"
 MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 ValueHistories = "98cad3c8-aec3-5f06-8e41-884608649ab7"
@@ -17,7 +18,8 @@ DecisionFocusedLearningAlgorithms = {path = ".."}
 Aqua = "0.8"
 DecisionFocusedLearningBenchmarks = "0.5"
 Documenter = "1"
-JuliaFormatter = "1"
+JuliaFormatter = "2"
+LoweredCodeUtils = "<3.6" # pin until https://github.com/aviatesk/JET.jl/issues/823 is fixed
 MLUtils = "0.4"
 Test = "1"
 ValueHistories = "0.5"