1 files changed, 8 insertions, 5 deletions
diff --git a/pkg/gui/popup/fake_popup_handler.go b/pkg/gui/popup/fake_popup_handler.go
index 95b0a3b1d..87e702668 100644
--- a/pkg/gui/popup/fake_popup_handler.go
+++ b/pkg/gui/popup/fake_popup_handler.go
@@ -1,6 +1,9 @@
 package popup
 
-import "github.com/jesseduffield/lazygit/pkg/gui/types"
+import (
+	"github.com/jesseduffield/gocui"
+	"github.com/jesseduffield/lazygit/pkg/gui/types"
+)
 
 type FakePopupHandler struct {
 	OnErrorMsg func(message string) error
@@ -30,12 +33,12 @@ func (self *FakePopupHandler) Prompt(opts types.PromptOpts) error {
 	return self.OnPrompt(opts)
 }
 
-func (self *FakePopupHandler) WithLoaderPanel(message string, f func() error) error {
-	return f()
+func (self *FakePopupHandler) WithLoaderPanel(message string, f func(*gocui.Task) error) error {
+	return f(&gocui.Task{})
 }
 
-func (self *FakePopupHandler) WithWaitingStatus(message string, f func() error) error {
-	return f()
+func (self *FakePopupHandler) WithWaitingStatus(message string, f func(*gocui.Task) error) error {
+	return f(&gocui.Task{})
 }
 
 func (self *FakePopupHandler) Menu(opts types.CreateMenuOptions) error {