diff --git a/ai21/clients/sagemaker/resources/sagemaker_completion.py b/ai21/clients/sagemaker/resources/sagemaker_completion.py
index 377cc4bd..97a9682a 100644
--- a/ai21/clients/sagemaker/resources/sagemaker_completion.py
+++ b/ai21/clients/sagemaker/resources/sagemaker_completion.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import List, Dict
 
 from ai21.clients.sagemaker.resources.sagemaker_resource import SageMakerResource