diff --git a/samples/snippets/src/test/java/com/example/firestore/beam/ExampleFirestoreWriteReadTest.java b/samples/snippets/src/test/java/com/example/firestore/beam/ExampleFirestoreWriteReadTest.java
index bd94fcf817..04d5a5662d 100644
--- a/samples/snippets/src/test/java/com/example/firestore/beam/ExampleFirestoreWriteReadTest.java
+++ b/samples/snippets/src/test/java/com/example/firestore/beam/ExampleFirestoreWriteReadTest.java
@@ -55,7 +55,9 @@ public void setUpStream() {
 
   @AfterClass
   public static void tearDown() {
-    deleteCollection(firestore.collection(collectionId), 1);
+    if (firestore != null && collectionId != null) {
+      deleteCollection(firestore.collection(collectionId), 1);
+    }
   }
 
   @Test