diff --git a/scripts/extension/attributes/morphology_impl.py b/scripts/extension/attributes/morphology_impl.py
index da9cf6a1aa3406d5da8d84f2ae7b244201438d74..acf192d8fb8d3ff088f4cea01ab97074cffceed4 100644
--- a/scripts/extension/attributes/morphology_impl.py
+++ b/scripts/extension/attributes/morphology_impl.py
@@ -1,13 +1,10 @@
 from datetime import datetime
 
-# TODO - uncomment this part
 # this is a task called by multiple processes,
 # so we need to restrict the number of threads used by numpy
-# from cluster_tools.utils.numpy_utils import set_numpy_threads
-# set_numpy_threads(1)
-from tmp.numpy_utils import set_numpy_threads
-
+from cluster_tools.utils.numpy_utils import set_numpy_threads
 set_numpy_threads(1)
+
 import numpy as np
 
 import vigra