diff --git a/miplearn/solvers/internal.py b/miplearn/solvers/internal.py
index 353cc30..f3335d4 100644
--- a/miplearn/solvers/internal.py
+++ b/miplearn/solvers/internal.py
@@ -4,7 +4,9 @@
 
 import logging
 from abc import ABC, abstractmethod
-from typing import TypedDict, Callable, Any, Dict, List
+from typing import Callable, Any, Dict, List
+
+from typing_extensions import TypedDict
 
 from ..instance import Instance