fix display code