diff --git a/tests/functional/adapter/test_simple_snapshot.py b/tests/functional/adapter/test_simple_snapshot.py
index 879c7b9..76efc54 100644
--- a/tests/functional/adapter/test_simple_snapshot.py
+++ b/tests/functional/adapter/test_simple_snapshot.py
@@ -2,6 +2,9 @@
     BaseSimpleSnapshot,
     BaseSnapshotCheck,
 )
+from dbt.tests.adapter.simple_snapshot.new_record_mode import (
+    SnapshotNewRecordMode,
+)
 
 
 class TestSnapshot(BaseSimpleSnapshot):
@@ -10,3 +13,7 @@ class TestSnapshot(BaseSimpleSnapshot):
 
 class TestSnapshotCheck(BaseSnapshotCheck):
     pass
+
+
+class TestPostgresSnapshotNewRecordMode(SnapshotNewRecordMode):
+    pass