diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py
index 302d8550a2071af9544ce2044070b64c8a846eff..4af23c9b3d4a60e09edccec0801cd53465df0419 100644
--- a/pes_to_spec/model.py
+++ b/pes_to_spec/model.py
@@ -21,6 +21,7 @@ from sklearn.model_selection import train_test_split
 from sklearn.base import clone, MetaEstimatorMixin
 from joblib import Parallel, delayed
 from functools import partial
+import multiprocessing as mp
 
 from typing import Any, Dict, List, Optional, Union, Tuple