diff --git a/airbyte_cdk/models/airbyte_protocol.py b/airbyte_cdk/models/airbyte_protocol.py
index 2528f7d7e..eaa674be9 100644
--- a/airbyte_cdk/models/airbyte_protocol.py
+++ b/airbyte_cdk/models/airbyte_protocol.py
@@ -2,6 +2,8 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 #
 
+from __future__ import annotations
+
 from dataclasses import InitVar, dataclass
 from typing import Annotated, Any, Dict, List, Mapping, Optional, Union
 
diff --git a/airbyte_cdk/models/file_transfer_record_message.py b/airbyte_cdk/models/file_transfer_record_message.py
index dcc1b7a92..ed1aad1a0 100644
--- a/airbyte_cdk/models/file_transfer_record_message.py
+++ b/airbyte_cdk/models/file_transfer_record_message.py
@@ -1,5 +1,7 @@
 # Copyright (c) 2024 Airbyte, Inc., all rights reserved.
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict, Optional