Fix test to submit job to correct partition