Fix notification calling the wrong class