diff --git a/src/tui/tui.zig b/src/tui/tui.zig
index b30b9a9..eee435f 100644
--- a/src/tui/tui.zig
+++ b/src/tui/tui.zig
@@ -1409,6 +1409,11 @@ pub fn get_keybind_mode() ?Mode {
     return self.input_mode_ orelse self.delayed_init_input_mode;
 }
 
+pub fn set_drag_source(drag_source: *Widget) void {
+    const self = current();
+    self.drag_source = drag_source;
+}
+
 pub fn reset_drag_context() void {
     const self = current();
     self.drag_source = null;