diff --git a/src/test/java/io/debezium/connector/db2/SourceInfoTest.java b/src/test/java/io/debezium/connector/db2/SourceInfoTest.java
index 38dede7..667d410 100644
--- a/src/test/java/io/debezium/connector/db2/SourceInfoTest.java
+++ b/src/test/java/io/debezium/connector/db2/SourceInfoTest.java
@@ -92,8 +92,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("ts_us", Schema.OPTIONAL_INT64_SCHEMA)
+                .field("ts_ns", Schema.OPTIONAL_INT64_SCHEMA)
                 .field("schema", Schema.STRING_SCHEMA)
                 .field("table", Schema.STRING_SCHEMA)
                 .field("change_lsn", Schema.OPTIONAL_STRING_SCHEMA)