diff --git a/lua/ivy/controller.lua b/lua/ivy/controller.lua
index be93ef6..0fe3b5c 100644
--- a/lua/ivy/controller.lua
+++ b/lua/ivy/controller.lua
@@ -37,7 +37,9 @@ controller.update = function(text)
 end
 
 controller.complete = function()
-  controller.checkpoint()
+  vim.api.nvim_set_current_win(window.origin)
+  controller.callback(window.get_current_selection())
+
   controller.destroy()
 end