lib/fasten/runner.rb in fasten-0.12.0 vs lib/fasten/runner.rb in fasten-0.12.2

@@ -114,10 +114,10 @@
         ui.force_clear
       end
     end
 
     def should_wait_for_running_tasks?
-      tasks.running? && (tasks.no_waiting? || tasks.failed? || %i[PAUSING QUITTING].include?(state)) || tasks.running.count >= jobs
+      tasks.running? && (tasks.no_waiting? || tasks.failed? || %i[PAUSING QUITTING].include?(state)) || tasks.running.map(&:weight).sum >= jobs
     end
 
     def wait_for_running_tasks
       use_threads ? wait_for_running_tasks_thread : wait_for_running_tasks_fork
     end