diff --git a/layer_shell.c b/layer_shell.c
index 5fe9b3b..a9b70bc 100644
--- a/layer_shell.c
+++ b/layer_shell.c
@@ -371,4 +371,4 @@ void nedm_arrange_layers(struct nedm_output *output) {
 	arrange_layer(output, 2, &full_area, &usable_area, false); // TOP
 	arrange_layer(output, 1, &full_area, &usable_area, false); // BOTTOM
 	arrange_layer(output, 0, &full_area, &usable_area, false); // BACKGROUND
-}
\ No newline at end of file
+}
diff --git a/output.c b/output.c
index aa06c5e..874eb85 100644
--- a/output.c
+++ b/output.c
@@ -678,6 +678,7 @@ handle_new_output(struct wl_listener *listener, void *data) {
 	output->layers[3] = wlr_scene_tree_create(&output->scene_output->scene->tree); // ZWLR_LAYER_SHELL_V1_LAYER_OVERLAY
 
 	output->wlr_output = wlr_output;
+  wlr_output->data = output;
 	output->destroyed = false;
 	output->wallpaper = NULL;
 	wl_signal_init(&output->events.destroy);