Fix compile

Author: schmidt-sebastian

packages/firestore/lite/src/api/transaction.ts

@@ -178,15 +178,14 @@ export function runTransaction<T>(
   updateFunction: (transaction: firestore.Transaction) => Promise<T>
 ): Promise<T> {
   const firestoreClient = cast(firestore, Firestore);
-  return getDatastore(firestoreClient).then(async datastore => {
-    const deferred = new Deferred<T>();
-    new TransactionRunner<T>(
-      new AsyncQueue(),
-      datastore,
-      internalTransaction =>
-        updateFunction(new Transaction(firestoreClient, internalTransaction)),
-      deferred
-    ).run();
-    return deferred.promise;
-  });
+  const datastore = getDatastore(firestoreClient);
+  const deferred = new Deferred<T>();
+  new TransactionRunner<T>(
+    new AsyncQueue(),
+    datastore,
+    internalTransaction =>
+      updateFunction(new Transaction(firestoreClient, internalTransaction)),
+    deferred
+  ).run();
+  return deferred.promise;
 }

packages/firestore/lite/src/api/write_batch.ts

@@ -181,9 +181,8 @@ export function writeBatch(
   firestore: firestore.FirebaseFirestore
 ): firestore.WriteBatch {
   const firestoreImpl = cast(firestore, Firestore);
+  const datastore = getDatastore(firestoreImpl);
   return new WriteBatch(firestoreImpl, writes =>
-    getDatastore(firestoreImpl).then(datastore =>
-      invokeCommitRpc(datastore, writes)
-    )
+    invokeCommitRpc(datastore, writes)
   );
 }