1 files changed, 8 insertions, 6 deletions
diff --git a/Firestore/Example/Tests/Remote/FSTSerializerBetaTests.mm b/Firestore/Example/Tests/Remote/FSTSerializerBetaTests.mm
index c0e9cad..454b108 100644
--- a/Firestore/Example/Tests/Remote/FSTSerializerBetaTests.mm
+++ b/Firestore/Example/Tests/Remote/FSTSerializerBetaTests.mm
@@ -50,6 +50,7 @@
 #include "Firestore/core/src/firebase/firestore/model/database_id.h"
 #include "Firestore/core/src/firebase/firestore/model/field_mask.h"
 #include "Firestore/core/src/firebase/firestore/model/field_transform.h"
+#include "Firestore/core/src/firebase/firestore/model/precondition.h"
 #include "Firestore/core/src/firebase/firestore/util/string_apple.h"
 #include "Firestore/core/test/firebase/firestore/testutil/testutil.h"
 
@@ -58,6 +59,7 @@ namespace util = firebase::firestore::util;
 using firebase::firestore::model::DatabaseId;
 using firebase::firestore::model::FieldMask;
 using firebase::firestore::model::FieldTransform;
+using firebase::firestore::model::Precondition;
 
 NS_ASSUME_NONNULL_BEGIN
 
@@ -377,12 +379,12 @@ NS_ASSUME_NONNULL_BEGIN
 }
 
 - (void)testEncodesSetMutationWithPrecondition {
-  FSTSetMutation *mutation = [[FSTSetMutation alloc]
-       initWithKey:FSTTestDocKey(@"foo/bar")
-             value:FSTTestObjectValue(
-                       @{ @"a" : @"b",
-                          @"num" : @1 })
-      precondition:[FSTPrecondition preconditionWithUpdateTime:FSTTestVersion(4)]];
+  FSTSetMutation *mutation =
+      [[FSTSetMutation alloc] initWithKey:FSTTestDocKey(@"foo/bar")
+                                    value:FSTTestObjectValue(
+                                              @{ @"a" : @"b",
+                                                 @"num" : @1 })
+                             precondition:Precondition::UpdateTime(testutil::Version(4))];
   GCFSWrite *proto = [GCFSWrite message];
   proto.update = [self.serializer encodedDocumentWithFields:mutation.value key:mutation.key];
   proto.currentDocument.updateTime =