diff --git a/airbyte_cdk/test/entrypoint_wrapper.py b/airbyte_cdk/test/entrypoint_wrapper.py
index c6e4dd9e4..3aecbf7a5 100644
--- a/airbyte_cdk/test/entrypoint_wrapper.py
+++ b/airbyte_cdk/test/entrypoint_wrapper.py
@@ -65,6 +65,11 @@ class AirbyteEntrypointException(Exception):
             raise output.as_exception()
     """
 
+    message: str = ""
+
+    def __post_init__(self) -> None:
+        super().__init__(self.message)
+
 
 class EntrypointOutput:
     """A class to encapsulate the output of an Airbyte connector's execution.