diff --git a/tests/test_persistence.py b/tests/test_persistence.py
index 82e6c0e..a3da1fa 100644
--- a/tests/test_persistence.py
+++ b/tests/test_persistence.py
@@ -42,7 +42,7 @@ def test_migration():
         },
     }
     result = migration.migrate(v1)
-    assert result["version"] == 2
+    assert result["version"] == content.SAVE_VERSION
     assert result["cat"]["fullness"] == 100
     assert "last_updated" in result["cat"]