diff --git a/scripts/db_migrator.py b/scripts/db_migrator.py
index 734ef288ea..32731fac49 100755
--- a/scripts/db_migrator.py
+++ b/scripts/db_migrator.py
@@ -857,8 +857,8 @@ def version_4_0_0(self):
         # reading FAST_REBOOT table can't be done with stateDB.get as it uses hget behind the scenes and the table structure is
         # not using hash and won't work.
         # FAST_REBOOT table exists only if fast-reboot was triggered.
-        keys = self.stateDB.keys(self.stateDB.STATE_DB, "FAST_REBOOT")
-        if keys is not None:
+        keys = self.stateDB.keys(self.stateDB.STATE_DB, "FAST_REBOOT|system")
+        if keys:
             enable_state = 'true'
         else:
             enable_state = 'false'