diff --git a/NetMedPy/NetMedPy.py b/NetMedPy/NetMedPy.py
index 1f27551..1ce3b79 100644
--- a/NetMedPy/NetMedPy.py
+++ b/NetMedPy/NetMedPy.py
@@ -78,7 +78,7 @@
 import networkx as nx
 import numpy as np
 import pickle
-from multiprocessing import cpu_countt
+from multiprocessing import cpu_count
 import random
 import ray
 import warnings