diff --git a/pkg/gui/commit_files_panel.go b/pkg/gui/commit_files_panel.go
index d34d1a14d..4b9833c44 100644
--- a/pkg/gui/commit_files_panel.go
+++ b/pkg/gui/commit_files_panel.go
@@ -94,7 +94,8 @@ func (gui *Gui) handleDiscardOldFileChange() error {
 }
 
 func (gui *Gui) refreshCommitFilesView() error {
-	if gui.currentSideContext().GetKey() == COMMIT_FILES_CONTEXT_KEY {
+	currentSideContext := gui.currentSideContext()
+	if currentSideContext.GetKey() == COMMIT_FILES_CONTEXT_KEY || currentSideContext.GetKey() == BRANCH_COMMITS_CONTEXT_KEY {
 		if err := gui.handleRefreshPatchBuildingPanel(-1); err != nil {
 			return err
 		}