diff --git a/tests/integration/test_webhook.py b/tests/integration/test_webhook.py index 7d26fd4f..b5fae924 100644 --- a/tests/integration/test_webhook.py +++ b/tests/integration/test_webhook.py @@ -14,6 +14,7 @@ ListOfRuns, ListOfWebhookDispatches, ListOfWebhooks, + Run, Webhook, WebhookDispatch, WebhookEventType, @@ -39,7 +40,7 @@ async def _get_finished_run_id(client: ApifyClient | ApifyClientAsync) -> str: # No completed runs found - start one and wait for it to finish run = await maybe_await(client.actor(HELLO_WORLD_ACTOR).call()) - assert isinstance(run, ListOfRuns) + assert isinstance(run, Run) return run.id