Fix for notification box