Ensure we use the right topology for the stage