diff --git a/turbo_alignment/dataset/chat/chat.py b/turbo_alignment/dataset/chat/chat.py
index f1ac605..5c5e09a 100755
--- a/turbo_alignment/dataset/chat/chat.py
+++ b/turbo_alignment/dataset/chat/chat.py
@@ -2,7 +2,8 @@
 from abc import ABC
 from itertools import accumulate
 from pathlib import Path
-from typing import Any, Self, overload
+from typing import Any, overload
+from typing_extensions import Self
 
 import numpy as np
 import numpy.typing as npt