fix the bug for display