1 files changed, 2 insertions, 2 deletions

diff --git a/sway/desktop/output.c b/sway/desktop/output.c
index 6ddcac3b..27579c1b 100644
--- a/sway/desktop/output.c
+++ b/sway/desktop/output.c
@@ -5,6 +5,7 @@
 #include <wlr/types/wlr_output.h>
 #include <wlr/render.h>
 #include "log.h"
+#include "sway/container.h"
 #include "sway/output.h"
 #include "sway/server.h"
 
@@ -34,11 +35,10 @@ void output_add_notify(struct wl_listener *listener, void *data) {
         struct sway_output *output = calloc(1, sizeof(struct sway_output));
         output->wlr_output = wlr_output;
         output->server = server;
+        output->swayc = new_output(output);
 
         output->frame.notify = output_frame_notify;
         wl_signal_add(&wlr_output->events.frame, &output->frame);
-
-        // TODO: Add to tree
 }
 
 void output_remove_notify(struct wl_listener *listener, void *data) {