diff --git a/tests/test_utils_tqdm_mod.py b/tests/test_utils_tqdm_mod.py
index 6a8ecf837..ed5025797 100644
--- a/tests/test_utils_tqdm_mod.py
+++ b/tests/test_utils_tqdm_mod.py
@@ -11,12 +11,12 @@ sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')
 from slic.utils.tqdm_mod import *
 
 
-def extract_last_line(output):
-    return output.strip().splitlines()[-1]
+def extract_lines(output, label):
+    return [line for line in output.strip().splitlines() if label in line]
 
 def get_bar_visual(line):
     try:
-        return line.split("|")[1]
+        return line.split("|")[1] 
     except IndexError:
         return ""