diff --git a/src/amd/compiler/tests/check_output.py b/src/amd/compiler/tests/check_output.py
index 84585e9b301..660431423e7 100644
--- a/src/amd/compiler/tests/check_output.py
+++ b/src/amd/compiler/tests/check_output.py
@@ -221,7 +221,14 @@ def do_match(g, pattern, output, skip_lines, in_func=False):
         c = pattern.read(1)
         fail = False
         if c == '':
-            break
+            if not in_func:
+                while output.peek() in [' ', '\t']:
+                    output.read(1)
+                if output.read(1) not in ['', '\n']:
+                    res.fail('expected end of output')
+
+            if res.success:
+               break
         elif output.peek() == '':
             res.fail('unexpected end of output')
         elif c == '\\':
@@ -310,14 +317,6 @@ def do_match(g, pattern, output, skip_lines, in_func=False):
 
         escape = False
 
-    if not in_func:
-        while output.peek() in [' ', '\t']:
-            output.read(1)
-
-        if output.read(1) not in ['', '\n']:
-            res.fail('expected end of output')
-            return res
-
     return res
 
 class PatternCheck(Check):
diff --git a/src/amd/compiler/tests/test_d3d11_derivs.cpp b/src/amd/compiler/tests/test_d3d11_derivs.cpp
index cb6f891e868..3906d7295e3 100644
--- a/src/amd/compiler/tests/test_d3d11_derivs.cpp
+++ b/src/amd/compiler/tests/test_d3d11_derivs.cpp
@@ -115,9 +115,8 @@ BEGIN_TEST(d3d11_derivs.discard)
    pbld.add_vsfs(vs, fs);
 
    /* The discard gets emitted as demote_if. */
-   //>> s2: %global,  s1: (kill)%_:scc = s_andn2_b64 (kill)%_, (kill)%_
-   //>> s2: %_:exec,  s1: (kill)%_:scc = s_wqm_b64 %global
-   //>> p_exit_early_if_not %_:exec
+   //>> s2: %_:exec,  s1: (kill)%_:scc = s_wqm_b64 %_
+   //! p_exit_early_if_not %_:exec
    //>> v4: %_ = image_sample (kill)%_, (kill)%_, v1: undef, (kill)%_, (kill)%_ 2d
    pbld.print_ir(VK_SHADER_STAGE_FRAGMENT_BIT, "ACO IR");
 END_TEST