fix display fn bug