diff --git a/debezium-connector-postgres/src/test/java/io/debezium/connector/postgresql/PostgresConnectorIT.java b/debezium-connector-postgres/src/test/java/io/debezium/connector/postgresql/PostgresConnectorIT.java
index 0280ab601..25f9c60f8 100644
--- a/debezium-connector-postgres/src/test/java/io/debezium/connector/postgresql/PostgresConnectorIT.java
+++ b/debezium-connector-postgres/src/test/java/io/debezium/connector/postgresql/PostgresConnectorIT.java
@@ -1625,6 +1625,7 @@ public void shouldOutputRecordsInCloudEventsFormat() throws Exception {
 
         start(PostgresConnector.class, configBuilder.build());
         assertConnectorIsRunning();
+        waitForSnapshotToBeCompleted();
 
         SourceRecords snapshotRecords = consumeRecordsByTopic(2);
         List<SourceRecord> snapshot = snapshotRecords.allRecordsInOrder();
@@ -1636,6 +1637,7 @@ public void shouldOutputRecordsInCloudEventsFormat() throws Exception {
         }
 
         // insert some more records and test streaming
+        waitForStreamingRunning();
         TestHelper.execute(INSERT_STMT);
 
         Testing.Print.enable();