diff --git a/DRAMSys/traceAnalyzer/scripts/tests.py b/DRAMSys/traceAnalyzer/scripts/tests.py
index 27cb71a5..754bb37f 100755
--- a/DRAMSys/traceAnalyzer/scripts/tests.py
+++ b/DRAMSys/traceAnalyzer/scripts/tests.py
@@ -269,9 +269,9 @@ def phase_transitions_are_valid(connection):
 
         validTransitions['PDNAB'] = set(['PRE', 'RD', 'RDA', 'WR', 'WRA', 'REFB'])
         validTransitions['PDNPB'] = set(['ACT', 'REFB', 'SREFB'])
-        validTransitions['SREFB'] = set(['ACT'])
+        validTransitions['SREFB'] = set(['ACT', 'REFB'])
     else:
-        validTransitions['PRE'] = set(['ACT', 'PRE_ALL'])
+        validTransitions['PRE'] = set(['ACT', 'PRE_ALL', 'REFA'])
         validTransitions['PRE_ALL'] = set(['REFA', 'SREF'])
         validTransitions['ACT'] = set(['RD', 'RDA', 'WR', 'WRA', 'PRE_ALL'])