Fix nil class issue of notification