diff --git a/src/primaite/game/agent/rewards.py b/src/primaite/game/agent/rewards.py
index cb8f8cb1..30baad6f 100644
--- a/src/primaite/game/agent/rewards.py
+++ b/src/primaite/game/agent/rewards.py
@@ -111,9 +111,9 @@ class DatabaseFileIntegrity(AbstractReward):
         """
         database_file_state = access_from_nested_dict(state, self.location_in_state)
         health_status = database_file_state["health_status"]
-        if health_status == "corrupted":
+        if health_status == 3:
             return -1
-        elif health_status == "good":
+        elif health_status == 1:
             return 1
         else:
             return 0