Small fix for notifications