fix possible display bug