diff --git a/tests/test_bench_registry.py b/tests/test_bench_registry.py
index b6687de..c222f56 100644
--- a/tests/test_bench_registry.py
+++ b/tests/test_bench_registry.py
@@ -12,6 +12,9 @@ EXPECTED_NAMES = [
     "gpt3-qkv",
     "ipcq-allreduce",
     "matmul-composite",
+    "milestone-1h-ccl",
+    "milestone-1h-gemm",
+    "milestone-gqa-headline",
     "qkv-gemm",
     "qkv-gemm-multi-pe",
     "va-offset-verify",
diff --git a/tests/test_memory_store.py b/tests/test_memory_store.py
index 45dc64b..abee241 100644
--- a/tests/test_memory_store.py
+++ b/tests/test_memory_store.py
@@ -58,10 +58,12 @@ def test_reshape():
 
 
 def test_shape_mismatch_raises():
+    # A read larger than the stored region is rejected. (Same-size reshapes
+    # are byte-conserving and allowed; (5,5)=100B exceeds the 48B region.)
     store = MemoryStore()
     data = np.arange(12, dtype=np.float32)
     store.write("tcm", 0x0, data)
-    with pytest.raises(ValueError, match="Shape mismatch"):
+    with pytest.raises(ValueError, match="Out-of-bounds read"):
         store.read("tcm", 0x0, shape=(5, 5))