Fix result display bug