diff --git a/remotefilelog/remotefilelogserver.py b/remotefilelog/remotefilelogserver.py
--- a/remotefilelog/remotefilelogserver.py
+++ b/remotefilelog/remotefilelogserver.py
@@ -59,7 +59,8 @@
     onetimesetup(ui)
 
     # don't send files to shallow clients during pulls
-    def generatefiles(orig, self, changedfiles, linknodes, commonrevs, source):
+    def generatefiles(orig, self, changedfiles, linknodes, commonrevs, source,
+                      *args, **kwargs):
         caps = self._bundlecaps or []
         if shallowrepo.requirement in caps:
             # only send files that don't match the specified patterns
@@ -77,7 +78,8 @@
                     includepattern, excludepattern)
 
             changedfiles = list([f for f in changedfiles if not m(f)])
-        return orig(self, changedfiles, linknodes, commonrevs, source)
+        return orig(self, changedfiles, linknodes, commonrevs, source,
+                    *args, **kwargs)
 
     if util.safehasattr(changegroup, 'cgpacker'):
         wrapfunction(changegroup.cgpacker, 'generatefiles', generatefiles)