diff --git a/test/runtests.jl b/test/runtests.jl
index 7997acb..a9be9d5 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -20,7 +20,7 @@ using Test
             z = y * (y[-1] + 0.5 * z) * θ + x[-1]
             y = z[-2] * x * b
         end
-        replace_worked = @test_warn "Symbols [:b] are not in variables or parameters" Consistent.replace_vars(
+        replace_worked = @test_logs (:warn, "Symbols [:b] are not in variables or parameters") Consistent.replace_vars(
                 test_eqs.args[[2, 4]], [:z, :y], Symbol[:x], [:θ]
             ) == [
                 :(endos[1] = endos[2] * (lags[2, end - 0] + 0.5 * endos[1]) * params[1] + exos[1, end + -1]),
@@ -45,7 +45,7 @@ using Test
         let eqs = @equations begin
                 Y = C
             end
-            @test_warn "Symbols [:C] are not in variables or parameters" model(
+            @test_logs (:warn, "Symbols [:C] are not in variables or parameters") model(
                 endos = @variables(Y),
                 exos = @variables(),
                 params = @variables(),