diff --git a/src/ui_server/mod.rs b/src/ui_server/mod.rs
index 67ab85e..24d4ab1 100644
--- a/src/ui_server/mod.rs
+++ b/src/ui_server/mod.rs
@@ -20,11 +20,12 @@ use tokio_tungstenite::{
     connect_async_tls_with_config, tungstenite,
     tungstenite::protocol::Message as TungsteniteMessage,
 };
+#[cfg(not(feature = "embed-ui"))]
 use tower_http::services::{ServeDir, ServeFile};
 
 #[cfg(feature = "embed-ui")]
 #[derive(RustEmbed)]
-#[folder = "../ui"]
+#[folder = "ui"]
 struct Assets;
 
 use crate::shared::AppState;