diff --git a/pkg/auth/webapp/session_authflow.go b/pkg/auth/webapp/session_authflow.go
index 9bfe3fdba3..2917cd155f 100644
--- a/pkg/auth/webapp/session_authflow.go
+++ b/pkg/auth/webapp/session_authflow.go
@@ -362,7 +362,7 @@ func (TakeBranchResultSimple) takeBranchResult() {}
 type TakeBranchResultInputRetryHandler func(err error) (nextInput interface{})
 
 type TakeBranchResultInput struct {
-	Input                 interface{}
+	Input                 map[string]interface{}
 	NewAuthflowScreenFull func(flowResponse *authflow.FlowResponse, retriedForError error) *AuthflowScreenWithFlowResponse
 	OnRetry               *TakeBranchResultInputRetryHandler
 }