diff --git a/src/Precompile.jl b/src/Precompile.jl
index c3c86f55d..a125f0ca5 100644
--- a/src/Precompile.jl
+++ b/src/Precompile.jl
@@ -1,16 +1,13 @@
 using PrecompileTools: @setup_workload, @compile_workload
 
 @setup_workload begin
-    @static if haskey(ENV, "REACTANT_TEST_GROUP")
-        return
-    end
+    initialize_dialect()
+    cpu = XLA.CPUClient()
     @compile_workload begin
-        initialize_dialect()
-        cpu = XLA.CPUClient()
         x = Reactant.ConcreteRArray(randn(Float64, 2, 2); client=cpu)
         @code_hlo optimize = false sum(x)
-        XLA.free_client(cpu)
-        deinitialize_dialect()
     end
+    XLA.free_client(cpu)
+    deinitialize_dialect()
     XLA.cpuclientcount[] = 0
 end