diff --git a/exasol/nb_connector/extension_wrapper_common.py b/exasol/nb_connector/extension_wrapper_common.py
index fb034c7..1953aab 100644
--- a/exasol/nb_connector/extension_wrapper_common.py
+++ b/exasol/nb_connector/extension_wrapper_common.py
@@ -1,3 +1,4 @@
+from __future__ import annotations
 from typing import Optional
 
 from exasol.nb_connector.connections import open_pyexasol_connection