diff --git a/setup.py b/setup.py
index a36fb87c..b1d194f8 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@ def calculate_version():
 
 setup(
     name='TPOT2',
-    python_requires='<3.12', #for configspace compatibility
+    python_requires='>=3.10, <3.12', #for configspace compatibility
     version=package_version,
     author='Pedro Ribeiro',
     packages=find_packages(),
diff --git a/tpot2/config/get_configspace.py b/tpot2/config/get_configspace.py
index abb3abb9..72eac9a6 100644
--- a/tpot2/config/get_configspace.py
+++ b/tpot2/config/get_configspace.py
@@ -71,6 +71,7 @@
 from sklearn.svm import SVC, SVR, LinearSVR, LinearSVC
 from lightgbm import LGBMClassifier, LGBMRegressor
 import sklearn
+import sklearn.calibration as calibration
 from sklearn.naive_bayes import GaussianNB, BernoulliNB, MultinomialNB
 from sklearn.decomposition import FastICA, PCA
 from sklearn.cluster import FeatureAgglomeration