Fixed tests

PostHog · Jul 25, 2024 · 02864eb · 02864eb
1 parent 7809c9d
commit 02864eb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/posthog/temporal/tests/batch_exports/test_import_data.py b/posthog/temporal/tests/batch_exports/test_import_data.py
@@ -87,6 +87,7 @@ async def test_postgres_source_without_ssh_tunnel(activity_environment, team, **
             table_names=["table_1"],
             incremental_field=None,
             incremental_field_type=None,
+            team_id=team.id,
         )
 
 
@@ -125,6 +126,7 @@ async def test_postgres_source_with_ssh_tunnel_disabled(activity_environment, te
             table_names=["table_1"],
             incremental_field=None,
             incremental_field_type=None,
+            team_id=team.id,
         )
 
 
@@ -180,4 +182,5 @@ def __exit__(self, exc_type, exc_value, exc_traceback):
             table_names=["table_1"],
             incremental_field=None,
             incremental_field_type=None,
+            team_id=team.id,
         )