diff --git a/lib/core/test/unit/Cardano/Wallet/Primitive/Types/TokenMapSpec.hs b/lib/core/test/unit/Cardano/Wallet/Primitive/Types/TokenMapSpec.hs
index b1f4f7c3f27..cd21d8ecfbb 100644
--- a/lib/core/test/unit/Cardano/Wallet/Primitive/Types/TokenMapSpec.hs
+++ b/lib/core/test/unit/Cardano/Wallet/Primitive/Types/TokenMapSpec.hs
@@ -452,8 +452,8 @@ testZeroValuedTokenQuantityFlat =
     token = dummyTokenName "DUMMY-TOKEN"
     json =
         [aesonQQ|
-          [ { "policy": #{policy}
-            , "token": #{token}
+          [ { "policy_id": #{policy}
+            , "asset_name": #{token}
             , "quantity": 0
             }
           ]
@@ -475,8 +475,8 @@ testZeroValuedTokenQuantityNested =
     token = dummyTokenName "DUMMY-TOKEN"
     json =
         [aesonQQ|
-          [ { "policy": #{policy}
-            , "tokens": [{"token": #{token}, "quantity": 0}]
+          [ { "policy_id": #{policy}
+            , "tokens": [{"asset_name": #{token}, "quantity": 0}]
             }
           ]
         |]
@@ -494,7 +494,7 @@ testEmptyTokenList =
         Left message
   where
     policy = dummyTokenPolicyId 'A'
-    json = [aesonQQ|[{"policy": #{policy}, "tokens": []}]|]
+    json = [aesonQQ|[{"policy_id": #{policy}, "tokens": []}]|]
     message = unwords
         [ failurePreamble
         , "Encountered empty token list for policy"