diff --git a/test/runtests.jl b/test/runtests.jl
index b64be58e..c45c641b 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -2930,7 +2930,7 @@ end
         m = @which show([1,2,3])
         @test definition(m) isa Expr
         m = @which redirect_stdout()
-        @test definition(m).head === :function
+        @test definition(m).head ∈ (:function, :(=))
 
         # Tracking stdlibs
         Revise.track(Unicode)
diff --git a/test/sigtest.jl b/test/sigtest.jl
index a4e9ed47..f69e7bdc 100644
--- a/test/sigtest.jl
+++ b/test/sigtest.jl
@@ -1,4 +1,6 @@
-using Revise, Test, CodeTracking, LoweredCodeUtils
+using Revise, Test
+using Revise.CodeTracking
+using Revise.LoweredCodeUtils
 
 function isdefinedmod(mod::Module)
     # Not all modules---e.g., LibGit2---are reachable without loading the stdlib