Fix test to work with job arrays