Fix notification broken in last commit