diff --git a/tests/python/opinfo/opinfo_input_generators.py b/tests/python/opinfo/opinfo_input_generators.py
index d1291a1f944..6481ceed9ae 100644
--- a/tests/python/opinfo/opinfo_input_generators.py
+++ b/tests/python/opinfo/opinfo_input_generators.py
@@ -914,9 +914,7 @@ def topk_error_generator(
     ), RuntimeError, "Invalid resized domain extent"
 
     #  error coming from aten fallback.
-    yield SampleInput(
-        a, 16, 1, True, False
-    ), RuntimeError, "selected index k out of range"
+    yield SampleInput(a, 16, 1, True, False), RuntimeError, "k .* range"
 
 
 def index_select_generator(