Fix NPE in WorkloadTask