fix display problem