diff --git a/src/desktop/tray.rs b/src/desktop/tray.rs
index 7190809..1a7c5ee 100644
--- a/src/desktop/tray.rs
+++ b/src/desktop/tray.rs
@@ -6,6 +6,8 @@ use tauri::AppHandle;
 use tauri::tray::{TrayIcon, TrayIconBuilder};
 use tauri::menu::{Menu, MenuItem};
 
+use crate::desktop::safe_command::SafeCommand;
+
 #[derive(Clone)]
 pub struct TrayManager {
     hostname: Arc<RwLock<Option<String>>>,