1 files changed, 3 insertions, 10 deletions
diff --git a/src/widgets/EdgeDecoration.hpp b/src/widgets/EdgeDecoration.hpp
index f627636..0a977df 100644
--- a/src/widgets/EdgeDecoration.hpp
+++ b/src/widgets/EdgeDecoration.hpp
@@ -7,9 +7,9 @@
 #include <qwebengineprofile.h>
 #include <qwebengineview.h>
 
+#include "WebViewData.hpp"
 #include "utils.hpp"
 #include "widgets/WebView.hpp"
-#include "widgets/WebViewStack.hpp"
 
 class EdgeDecoration : public QWidget {
   Q_OBJECT
@@ -20,18 +20,11 @@ public:
   void set_size(uint16_t size_value);
   void set_html(const QString &content);
   void set_enabled(bool enabled_value);
-  void set_url(const QUrl &url) {
-    if (!webview.has_value())
-      return;
-    webview.value()->setUrl(url);
-  }
+  void set_url(const QUrl &url);
+  std::optional<WebViewId> get_view_id();
 
   DEFINE_GETTER(is_enabled, enabled)
 
-  std::optional<WebViewId> get_view_id() {
-    return webview.has_value() ? std::make_optional(webview.value()->get_id()) : std::nullopt;
-  }
-
 private:
   bool vertical;
   std::optional<WebView *> webview = std::nullopt;