diff --git a/Makefile b/Makefile
index 7d7b336..0927569 100644
--- a/Makefile
+++ b/Makefile
@@ -27,6 +27,7 @@ docs-dev:
 
 install-deps:
 	$(PIP) install -i https://pypi.gurobi.com gurobipy
+	$(PIP) install xpress
 	$(PIP) install -r requirements.txt
 
 install:
diff --git a/miplearn/types.py b/miplearn/types.py
index 4455af8..5223743 100644
--- a/miplearn/types.py
+++ b/miplearn/types.py
@@ -2,7 +2,9 @@
 #  Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
 #  Released under the modified BSD license. See COPYING.md for more details.
 
-from typing import TypedDict, Optional, Dict, Callable, Any
+from typing import Optional, Dict, Callable, Any
+
+from mypy_extensions import TypedDict
 
 TrainingSample = TypedDict(
     "TrainingSample",