diff --git a/posthog/temporal/batch_exports/s3_batch_export.py b/posthog/temporal/batch_exports/s3_batch_export.py index 3d57d55e9ce37..fb649407520ff 100644 --- a/posthog/temporal/batch_exports/s3_batch_export.py +++ b/posthog/temporal/batch_exports/s3_batch_export.py @@ -327,7 +327,7 @@ def continue_from_state(self, state: S3MultiPartUploadState): return self.upload_id - async def complete(self) -> str: + async def complete(self) -> str | None: if self.is_upload_in_progress() is False: raise NoUploadInProgressError() @@ -343,7 +343,7 @@ async def complete(self) -> str: self.upload_id = None self.parts = [] - return response["Location"] + return response.get("Key") async def abort(self): """Abort this S3 multi-part upload."""