diff --git a/src/Views/Widgets/WorkingCopy.xaml.cs b/src/Views/Widgets/WorkingCopy.xaml.cs
index affb8ff9..15bfb47a 100644
--- a/src/Views/Widgets/WorkingCopy.xaml.cs
+++ b/src/Views/Widgets/WorkingCopy.xaml.cs
@@ -120,7 +120,7 @@ namespace SourceGit.Views.Widgets {
 
             mergePanel.Visibility = Visibility.Collapsed;
 
-            if (change.IsAddedToIndex) {
+            if (!container.IsUnstaged) {
                 unstagedContainer.UnselectAll();
 
                 diffViewer.Diff(repo.Path, new DiffViewer.Option() {