diff --git a/firestore/cloud-client/requirements-test.txt b/firestore/cloud-client/requirements-test.txt
index 781d4326c94..1b569cb4f2c 100644
--- a/firestore/cloud-client/requirements-test.txt
+++ b/firestore/cloud-client/requirements-test.txt
@@ -1 +1,2 @@
 pytest==5.3.2
+flaky==3.6.1
diff --git a/firestore/cloud-client/snippets_test.py b/firestore/cloud-client/snippets_test.py
index 36d19877e65..908fdcd7fbe 100644
--- a/firestore/cloud-client/snippets_test.py
+++ b/firestore/cloud-client/snippets_test.py
@@ -267,12 +267,14 @@ def test_cursor_multiple_conditions():
     snippets.cursor_multiple_conditions()
 
 
+@pytest.mark.flaky(max_runs=3)
 def test_listen_document(capsys):
     snippets.listen_document()
     out, _ = capsys.readouterr()
     assert 'Received document snapshot: SF' in out
 
 
+@pytest.mark.flaky(max_runs=3)
 def test_listen_multiple(capsys):
     snippets.listen_multiple()
     out, _ = capsys.readouterr()
@@ -280,6 +282,7 @@ def test_listen_multiple(capsys):
     assert 'SF' in out
 
 
+@pytest.mark.flaky(max_runs=3)
 def test_listen_for_changes(capsys):
     snippets.listen_for_changes()
     out, _ = capsys.readouterr()