Fix js display bug