1 files changed, 8 insertions, 2 deletions
diff --git a/lib/gitlab/gitaly_client/diff_stitcher.rb b/lib/gitlab/gitaly_client/diff_stitcher.rb
index 98d327a7329..e98ae75590d 100644
--- a/lib/gitlab/gitaly_client/diff_stitcher.rb
+++ b/lib/gitlab/gitaly_client/diff_stitcher.rb
@@ -5,8 +5,10 @@ module Gitlab
     class DiffStitcher
       include Enumerable
 
-      def initialize(rpc_response)
-        @rpc_response = rpc_response
+      delegate :size, to: :rpc_response
+
+      def initialize(rpc_response_param)
+        @rpc_response = rpc_response_param
       end
 
       def each
@@ -31,6 +33,10 @@ module Gitlab
           end
         end
       end
+
+      private
+
+      attr_reader :rpc_response
     end
   end
 end