Neither `return` is necessary, since the whole function is just `if (!jobIdToStageIds.contains(jobId)) Set[Int]() else removedStages`.