Small fix to notifications