This isn't really related to your change but as long as you're changing things here, I don't think this comment is relevant anymore...or at least I don't think DAGScheduler requires that the port of this server is known?  So maybe just delete that line.