diff --git a/tests/functional/adapter/test_incremental_microbatch.py b/tests/functional/adapter/test_incremental_microbatch.py
index ce5855b..b9c8e8e 100644
--- a/tests/functional/adapter/test_incremental_microbatch.py
+++ b/tests/functional/adapter/test_incremental_microbatch.py
@@ -1,7 +1,11 @@
 from dbt.tests.adapter.incremental.test_incremental_microbatch import (
-    BaseMicrobatch,
+    BaseTestMicrobatchOn,
+    BaseTestMicrobatchOff
 )
 
 
-class TestPostgresMicrobatch(BaseMicrobatch):
+class TestPostgresMicrobatchOn(BaseTestMicrobatchOn):
+    pass
+
+class TestPostgresMicrobatchOff(BaseTestMicrobatchOff):
     pass