diff --git a/tools/checks/check-datasource-composition.py b/tools/checks/check-datasource-composition.py
index 0149de2..c30c17b 100644
--- a/tools/checks/check-datasource-composition.py
+++ b/tools/checks/check-datasource-composition.py
@@ -43,6 +43,8 @@ from govoplan_dataflow.backend.db.models import (
 from govoplan_dataflow.backend.service import create_pipeline, preview_pipeline
 from govoplan_datasources.backend.db.models import (
     DatasourceMaterializationRecord,
+    DatasourcePayloadRecord,
+    DatasourcePayloadRowRecord,
     DatasourcePublicationRecord,
     DatasourceRecord,
     DatasourceStageRecord,
@@ -62,6 +64,8 @@ def main() -> int:
         tables=[
             ConnectorTabularSource.__table__,
             DatasourceRecord.__table__,
+            DatasourcePayloadRecord.__table__,
+            DatasourcePayloadRowRecord.__table__,
             DatasourceMaterializationRecord.__table__,
             DatasourceStageRecord.__table__,
             DatasourcePublicationRecord.__table__,