diff --git a/test/nn/aggr/test_basic.py b/test/nn/aggr/test_basic.py
index a73d53e18d3e..24227ed25488 100644
--- a/test/nn/aggr/test_basic.py
+++ b/test/nn/aggr/test_basic.py
@@ -23,7 +23,7 @@ def test_validate():
     with pytest.raises(ValueError, match="invalid dimension"):
         aggr(x, index, dim=-3)
 
-    with pytest.raises(ValueError, match="mismatch between"):
+    with pytest.raises(ValueError, match="invalid 'dim_size'"):
         aggr(x, ptr=ptr, dim_size=2)