fix display bug
fix display bug better