diff --git a/src/test/regress/citus_tests/run_test.py b/src/test/regress/citus_tests/run_test.py
index 24a9d8b36..a34a6e0a7 100755
--- a/src/test/regress/citus_tests/run_test.py
+++ b/src/test/regress/citus_tests/run_test.py
@@ -354,6 +354,7 @@ def needed_worker_count(test_name, dependencies):
     worker_count = worker_count_for(test_name)
     for dependency in dependencies.extra_tests():
         worker_count = max(worker_count_for(dependency), worker_count)
+    return worker_count
 
 
 if __name__ == "__main__":