diff --git a/quantile_forest/tests/test_quantile_forest.py b/quantile_forest/tests/test_quantile_forest.py
index 7e0a6b3..9536d17 100755
--- a/quantile_forest/tests/test_quantile_forest.py
+++ b/quantile_forest/tests/test_quantile_forest.py
@@ -3,7 +3,7 @@
 import math
 import pickle
 import warnings
-from typing import Any, Dict
+from typing import Any
 
 import numpy as np
 import pytest
@@ -41,7 +41,7 @@
 X_california = california.data[perm]
 y_california = california.target[perm]
 
-FOREST_REGRESSORS: Dict[str, Any] = {
+FOREST_REGRESSORS: dict[str, Any] = {
     "ExtraTreesQuantileRegressor": ExtraTreesQuantileRegressor,
     "RandomForestQuantileRegressor": RandomForestQuantileRegressor,
 }