diff --git a/stemflow/model/AdaSTEM.py b/stemflow/model/AdaSTEM.py
index 215dc05..e6b7645 100644
--- a/stemflow/model/AdaSTEM.py
+++ b/stemflow/model/AdaSTEM.py
@@ -1070,6 +1070,11 @@ def predict_proba(
                 Return numpy.ndarray of shape (n_samples, n_ensembles)
 
         """
+        # Setup joblib_tmp_dir
+        self.joblib_tmp_dir = initiate_joblib_tmp_dir(self.lazy_loading_dir)
+        self.duckdb_config = duckdb_config(self.max_mem, self.joblib_tmp_dir)
+        
+        #
         check_X_test(X_test, self)
         check_prediction_aggregation(aggregation)
         return_by_separate_ensembles, return_std = check_prediction_return(return_by_separate_ensembles, return_std)
@@ -1077,10 +1082,6 @@ def predict_proba(
         n_jobs = check_transform_n_jobs(self, n_jobs)
         self.base_model_method = base_model_method
         self.base_model_prediction_param = base_model_prediction_param
-
-        # Setup joblib_tmp_dir
-        self.joblib_tmp_dir = initiate_joblib_tmp_dir(self.lazy_loading_dir)
-        self.duckdb_config = duckdb_config(self.max_mem, self.joblib_tmp_dir)
         
         try:
             # predict
diff --git a/stemflow/version.py b/stemflow/version.py
index 1436d8f..bf78826 100644
--- a/stemflow/version.py
+++ b/stemflow/version.py
@@ -1 +1 @@
-__version__ = "1.1.6"
+__version__ = "1.1.7"