diff --git a/rootston/main.c b/rootston/main.c
index 7b1ca298..65a06a2f 100644
--- a/rootston/main.c
+++ b/rootston/main.c
@@ -41,6 +41,7 @@ int main(int argc, char **argv) {
 		return 1;
 	}
 
+	setenv("WAYLAND_DISPLAY", socket, true);
 	wl_display_run(server.wl_display);
 	return 0;
 }