add protection for display