diff --git a/tests/test_worker.py b/tests/test_worker.py
index 9535658..2a1bed3 100644
--- a/tests/test_worker.py
+++ b/tests/test_worker.py
@@ -1,6 +1,6 @@
 import pytest
 
-from    thumb_gen.worker  import  Generator
+from thumb_gen.worker   import Generator
 
 def test_worker():
     #video_path, output_path='', custom_text='True', font_dir='', font_size=0
@@ -9,4 +9,4 @@ def test_worker():
     custom_text = "thumb gen"
     app = Generator(video_path, output_path, custom_text)
     test_worker_value = app.run()
-    assert test_worker_value == 1
+    assert test_worker_value == True
diff --git a/thumb_gen/worker.py b/thumb_gen/worker.py
index dadbf31..6391fd2 100644
--- a/thumb_gen/worker.py
+++ b/thumb_gen/worker.py
@@ -47,4 +47,4 @@ def run(self):
         if thumb_out:
             print_success(self.output_folder)
 
-        return 1
+        return True