diff -r 61e405fb6372 -r c22fd3c4c23e hgext/largefiles/proto.py --- a/hgext/largefiles/proto.py Wed Mar 28 15:09:34 2018 -0700 +++ b/hgext/largefiles/proto.py Wed Mar 28 10:52:40 2018 -0700 @@ -168,12 +168,13 @@ caps.append('largefiles=serve') return caps -def heads(repo, proto): +def heads(orig, repo, proto): '''Wrap server command - largefile capable clients will know to call lheads instead''' if lfutil.islfilesrepo(repo): return wireprototypes.ooberror(LARGEFILES_REQUIRED_MSG) - return wireproto.heads(repo, proto) + + return orig(repo, proto) def sshrepocallstream(self, cmd, **args): if cmd == 'heads' and self.capable('largefiles'):