Minor fix for the wormhole logic