Fix imports in notifications