diff --git a/test/zygote_type_stability.jl b/test/zygote_type_stability.jl
index 8a37d52c03..cd7af49d1e 100644
--- a/test/zygote_type_stability.jl
+++ b/test/zygote_type_stability.jl
@@ -1,4 +1,4 @@
-using Lux, Random, Zygote, StableRNGs, Test
+using Lux, Random, Zygote, StableRNGs, Test, TestExtras
 
 include("setup_modes.jl")