Is there a reason why you removed this check? If you look where `stageIdToDescription` is populated, it may or may contain an an entry for the stage.