diff --git a/DeepLabStream.py b/DeepLabStream.py
index 3d7c90a..9f58820 100755
--- a/DeepLabStream.py
+++ b/DeepLabStream.py
@@ -277,15 +277,15 @@ def get_pose_mp(input_q, output_q):
         """
 
         if MODEL_ORIGIN == 'DLC' or  MODEL_ORIGIN == 'MADLC':
+            from utils.poser import find_local_peaks_new
             config, sess, inputs, outputs = load_deeplabcut()
             while True:
                 if input_q.full():
                     index, frame = input_q.get()
                     if MODEL_ORIGIN == 'DLC':
                         scmap, locref, pose = get_pose(frame, config, sess, inputs, outputs)
-                        # TODO: Remove alterations to original
-                        #peaks = find_local_peaks_new(scmap, locref, ANIMALS_NUMBER, config)
-                        peaks = pose
+                        peaks = find_local_peaks_new(scmap, locref, ANIMALS_NUMBER, config)
+                        #peaks = pose
                     if MODEL_ORIGIN == 'MADLC':
                         peaks = get_ma_pose(frame, config, sess, inputs, outputs)