Fixed an issue where notifications weren't working