diff --git a/hyperglass/models/tests/test_util.py b/hyperglass/models/tests/test_util.py
index 9360033..ce103c6 100644
--- a/hyperglass/models/tests/test_util.py
+++ b/hyperglass/models/tests/test_util.py
@@ -9,8 +9,8 @@ from ..util import check_legacy_fields
 
 def test_check_legacy_fields():
     test1 = {"name": "Device A", "nos": "juniper"}
-    test1_expected = {"name": "Device A", "type": "juniper"}
-    test2 = {"name": "Device B", "type": "juniper"}
+    test1_expected = {"name": "Device A", "platform": "juniper"}
+    test2 = {"name": "Device B", "platform": "juniper"}
     test3 = {"name": "Device C"}
     assert set(check_legacy_fields("Device", **test1).keys()) == set(
         test1_expected.keys()