diff --git a/include/labwc.h b/include/labwc.h
index 39613272..71f74263 100644
--- a/include/labwc.h
+++ b/include/labwc.h
@@ -344,6 +344,8 @@ void cursor_init(struct seat *seat);
 
 void keyboard_init(struct seat *seat);
 
+void arrange_layers(struct output *output);
+
 void seat_init(struct server *server);
 void seat_finish(struct server *server);
 void seat_focus_surface(struct seat *seat, struct wlr_surface *surface);
diff --git a/src/server.c b/src/server.c
index 3c21eedd..4542264a 100644
--- a/src/server.c
+++ b/src/server.c
@@ -107,11 +107,10 @@ static void seat_disinhibit_input(struct seat *seat)
 	seat->active_client_while_inhibited = NULL;
 	// Triggers a refocus of the topmost surface layer if necessary
 	// TODO: Make layer surface focus per-output based on cursor position
-/*
-  struct roots_output *output;
-	wl_list_for_each(output, &seat->input->server->desktop->outputs, link) {
+
+	struct output *output;
+	wl_list_for_each(output, &seat->server->outputs, link)
 		arrange_layers(output);
-*/
 }