Fix notifications bug