diff --git a/dev/src/write-batch.ts b/dev/src/write-batch.ts
index 30916f2d2..2dae15609 100644
--- a/dev/src/write-batch.ts
+++ b/dev/src/write-batch.ts
@@ -16,7 +16,7 @@
 
 import * as assert from 'assert';
 
-import {google} from '../protos/firestore_proto_api';
+import {google} from '../protos/firestore_v1_proto_api';
 import {
   DocumentMask,
   DocumentSnapshot,