diff --git a/tests/proc/proc_rom.ys b/tests/proc/proc_rom.ys
index 93fd5002bd8..8b91a9021c2 100644
--- a/tests/proc/proc_rom.ys
+++ b/tests/proc/proc_rom.ys
@@ -33,7 +33,7 @@ hierarchy -auto-top
 
 design -save orig
 proc
-select -assert-count 1 t:$memrd_v2
+select -assert-count 1 t:$memrd_v2 a:src %i
 memory
 opt_dff
 design -stash postopt
@@ -80,7 +80,7 @@ hierarchy -auto-top
 
 design -save orig
 proc
-select -assert-count 1 t:$memrd_v2
+select -assert-count 1 t:$memrd_v2 a:src %i
 memory
 opt_dff
 design -stash postopt
@@ -127,7 +127,7 @@ hierarchy -auto-top
 
 design -save orig
 proc
-select -assert-count 1 t:$memrd_v2
+select -assert-count 1 t:$memrd_v2 a:src %i
 memory
 opt_dff
 design -stash postopt
@@ -175,7 +175,7 @@ hierarchy -auto-top
 
 design -save orig
 proc
-select -assert-count 1 t:$memrd_v2
+select -assert-count 1 t:$memrd_v2 a:src %i
 memory
 opt_dff
 design -stash postopt