diff --git a/parallelbar/parallelbar.py b/parallelbar/parallelbar.py
index 316d63d..aff143b 100644
--- a/parallelbar/parallelbar.py
+++ b/parallelbar/parallelbar.py
@@ -9,8 +9,7 @@
 from concurrent.futures import TimeoutError
 
 from tqdm.auto import tqdm
-from tools import get_len
-import time
+from .tools import get_len
 
 
 class ProgressBar(tqdm):
@@ -173,16 +172,3 @@ def progress_imapu(func, tasks, n_cpu=None, chunk_size=None, core_progress=False
                           disable)
     return result
 
-
-def foo(n):
-    if n == 50:
-        1 / 0
-    elif n == 10:
-        time.sleep(2)
-    else:
-        time.sleep(.5)
-    return n
-
-
-if __name__ == '__main__':
-    res = progress_map(foo, range(20), process_timeout=15)
diff --git a/setup.py b/setup.py
index 9b05ecc..5a282a3 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 
 setup(
     name='parallelbar',
-    version='0.2.5',
+    version='0.2.6',
     packages=find_packages(),
     author='Dubovik Pavel',
     author_email='geometryk@gmail.com',