Fixed dispay bug