fix bug for display