diff --git a/platform/vcs-impl/src/com/intellij/openapi/vcs/impl/PartialChangesUtil.java b/platform/vcs-impl/src/com/intellij/openapi/vcs/impl/PartialChangesUtil.java
index 1989c074b520..ce6e8b343ebf 100644
--- a/platform/vcs-impl/src/com/intellij/openapi/vcs/impl/PartialChangesUtil.java
+++ b/platform/vcs-impl/src/com/intellij/openapi/vcs/impl/PartialChangesUtil.java
@@ -56,16 +56,13 @@ public class PartialChangesUtil {
         if (change instanceof ChangeListChange) {
           ChangeListChange changelistChange = (ChangeListChange)change;
 
-          ContentRevision afterRevision = change.getAfterRevision();
-          if (afterRevision instanceof CurrentContentRevision) {
-            VirtualFile virtualFile = ((CurrentContentRevision)afterRevision).getVirtualFile();
-            if (virtualFile != null) {
-              partialChangesMap.putValue(virtualFile, changelistChange);
-              continue;
-            }
+          VirtualFile virtualFile = getVirtualFile(change);
+          if (virtualFile != null) {
+            partialChangesMap.putValue(virtualFile, changelistChange);
+          }
+          else {
+            otherChanges.add((changelistChange).getChange());
           }
-
-          otherChanges.add((changelistChange).getChange());
         }
         else {
           otherChanges.add(change);