Small notification fix