Fix the other check for same locality migration as well