Clean up the specs for dissociate_work_units a bit