diff --git a/src/test/java/io/debezium/connector/vitess/SourceInfoTest.java b/src/test/java/io/debezium/connector/vitess/SourceInfoTest.java
index 64a5702e..1948dffd 100644
--- a/src/test/java/io/debezium/connector/vitess/SourceInfoTest.java
+++ b/src/test/java/io/debezium/connector/vitess/SourceInfoTest.java
@@ -125,6 +125,8 @@ public void schemaIsCorrect() {
                 .field("snapshot", AbstractSourceInfoStructMaker.SNAPSHOT_RECORD_SCHEMA)
                 .field("db", Schema.STRING_SCHEMA)
                 .field("sequence", Schema.OPTIONAL_STRING_SCHEMA)
+                .field("ts_us", Schema.INT64_SCHEMA)
+                .field("ts_ns", Schema.INT64_SCHEMA)
                 .field("keyspace", Schema.STRING_SCHEMA)
                 .field("table", Schema.STRING_SCHEMA)
                 .field("shard", Schema.STRING_SCHEMA)