Update workload.rb

bug fix