diff --git a/tests/test_persistence.py b/tests/test_persistence.py
index 7d30cb3..34de4c5 100644
--- a/tests/test_persistence.py
+++ b/tests/test_persistence.py
@@ -2,8 +2,25 @@ from untitled import model, persistence
 
 
 def test_save_load_roundtrip(tmp_path):
-    original = model.Save(version=1, cat=model.Cat("Mittens"))
+    original = model.Save(
+        version=1,
+        cat=model.Cat(
+            "Fry",
+            {
+                "size": "tiny",
+                "color": "tuxedo",
+                "eyes": "blue",
+                "personality": "judges you silently",
+            },
+        ),
+    )
     persistence.save(original, tmp_path)
-    loaded = persistence.load("Mittens", tmp_path)
-    assert loaded.cat.name == "Mittens"
+    loaded = persistence.load("Fry", tmp_path)
+    assert loaded.cat.name == "Fry"
+    assert loaded.cat.traits == {
+        "size": "tiny",
+        "color": "tuxedo",
+        "eyes": "blue",
+        "personality": "judges you silently",
+    }
     assert loaded.version == 1
diff --git a/untitled/screens.py b/untitled/screens.py
index 1a2803a..766d2a0 100644
--- a/untitled/screens.py
+++ b/untitled/screens.py
@@ -1,4 +1,4 @@
-from untitled import generation, ui, rules, model
+from untitled import generation, model, rules, ui
 
 
 def adoption():