Fix revisit bug