1 files changed, 4 insertions, 2 deletions
diff --git a/Firestore/Example/Tests/Model/FSTMutationTests.mm b/Firestore/Example/Tests/Model/FSTMutationTests.mm
index 0bb7518..3183efd 100644
--- a/Firestore/Example/Tests/Model/FSTMutationTests.mm
+++ b/Firestore/Example/Tests/Model/FSTMutationTests.mm
@@ -145,7 +145,9 @@ using firebase::firestore::model::TransformOperation;
 - (void)testCreateArrayUnionTransform {
   FSTTransformMutation *transform = FSTTestTransformMutation(@"collection/key", @{
     @"foo" : [FIRFieldValue fieldValueForArrayUnion:@[ @"tag" ]],
-    @"bar.baz" : [FIRFieldValue fieldValueForArrayUnion:@[ @YES, @[ @1, @2 ], @{@"a" : @"b"} ]]
+    @"bar.baz" :
+        [FIRFieldValue fieldValueForArrayUnion:@[ @YES,
+                                                  @{ @"nested" : @{@"a" : @[ @1, @2 ]} } ]]
   });
   XCTAssertEqual(transform.fieldTransforms.size(), 2);
 
@@ -161,7 +163,7 @@ using firebase::firestore::model::TransformOperation;
   XCTAssertEqual(second.path(), FieldPath({"bar", "baz"}));
   {
     std::vector<FSTFieldValue *> expectedElements {
-      FSTTestFieldValue(@YES), FSTTestFieldValue(@[ @1, @2 ]), FSTTestFieldValue(@{@"a" : @"b"})
+      FSTTestFieldValue(@YES), FSTTestFieldValue(@{ @"nested" : @{@"a" : @[ @1, @2 ]} })
     };
     ArrayTransform expected(TransformOperation::Type::ArrayUnion, expectedElements);
     XCTAssertEqual(static_cast<const ArrayTransform &>(second.transformation()), expected);