fix test so that we no longer flatten when there's just one job