diff --git a/app/models/legacy_diff_note.rb b/app/models/legacy_diff_note.rb
index 25e90036a539f5d90784fc604d6d5906e394ea37..cfe09b338b2c089ffc2fc11de878b75de93ead73 100644
--- a/app/models/legacy_diff_note.rb
+++ b/app/models/legacy_diff_note.rb
@@ -32,7 +32,7 @@ def diff
   end
 
   def diff_file
-    @diff_file ||= Gitlab::Diff::File.new(diff, repository: project_repository) if diff
+    @diff_file ||= Gitlab::Diff::File.new(diff, repository: project_repository, diff_refs: noteable.diff_refs) if diff
   end
 
   def diff_line