diff --git a/src/ublue_update/cli.py b/src/ublue_update/cli.py
index e86144e..8d49124 100644
--- a/src/ublue_update/cli.py
+++ b/src/ublue_update/cli.py
@@ -6,7 +6,7 @@
 import logging
 import tomllib
 import argparse
-from notification_manager import NotificationManager
+from ublue_update.notification_manager import NotificationManager
 from gi.repository import GLib
 
 def check_cpu_load():