diff --git a/assets/batch_score/components/driver/src/batch_score_oss/aoai/scoring/aoai_scoring_client.py b/assets/batch_score/components/driver/src/batch_score_oss/aoai/scoring/aoai_scoring_client.py
index cf2ee339a5..a97b096b86 100644
--- a/assets/batch_score/components/driver/src/batch_score_oss/aoai/scoring/aoai_scoring_client.py
+++ b/assets/batch_score/components/driver/src/batch_score_oss/aoai/scoring/aoai_scoring_client.py
@@ -17,7 +17,7 @@ def __init__(
             self,
             header_provider: HeaderProvider,
             configuration: Configuration,
-            tally_handler: TallyFailedRequestHandler = None):
+            tally_handler: TallyFailedRequestHandler = None) -> None:
         """Initialize AoaiScoringClient."""
         super().__init__(
             header_provider=header_provider,