Fixed import for notifications