Update tests to mock notifications