diff --git a/src/Reactant.jl b/src/Reactant.jl
index 938b685cf..60e2d5f81 100644
--- a/src/Reactant.jl
+++ b/src/Reactant.jl
@@ -124,10 +124,8 @@ function set_default_backend(backend::String)
     return set_default_backend(XLA.backends[backend])
 end
 
-#HACK: Disable precompilation for test
-@info gethostname()
-@static if splitpath(Base.active_project())[3] != "runner"
-    @info "enable precompilation" Base.active_project()
+@static if !haskey(ENV, "REACTANT_TEST_GROUP")
+    @info "enable precompilation" gethostname() Base.active_project()
     include("precompile.jl")
 end