1 files changed, 11 insertions, 2 deletions

diff --git a/waltham-transmitter/transmitter-plugin/output.c b/waltham-transmitter/transmitter-plugin/output.c
index dc7d5d8..0676655 100644
--- a/waltham-transmitter/transmitter-plugin/output.c
+++ b/waltham-transmitter/transmitter-plugin/output.c
@@ -117,6 +117,14 @@ free_mode_list(struct wl_list *mode_list)
 	}
 }
 
+static int
+transmitter_output_disable(struct weston_output *base)
+{
+	struct weston_transmitter_output *output = wl_container_of(base, output, base);
+	wl_event_source_remove(output->finish_frame_timer);
+	return 0;
+}
+
 void
 transmitter_output_destroy(struct weston_transmitter_output *output)
 {
@@ -126,8 +134,8 @@ transmitter_output_destroy(struct weston_transmitter_output *output)
 	free_mode_list(&output->base.mode_list);
 	weston_head_release(head);
 	free(head);
-
-	weston_output_destroy(&output->base);
+	transmitter_output_disable(&output->base);
+	weston_output_release(&output->base);
 	free(output);
 }
 
@@ -342,6 +350,7 @@ transmitter_remote_create_output(struct weston_transmitter_remote *remote,
 	output->base.start_repaint_loop = transmitter_start_repaint_loop;
 	output->base.repaint = transmitter_output_repaint;
 	output->base.destroy = transmitter_output_destroy_;
+	output->base.disable = transmitter_output_disable;
 	output->base.assign_planes = NULL;
 	output->base.set_dpms = NULL;
 	output->base.switch_mode = NULL;