--- a/mercurial/wireproto.py	Fri Mar 28 14:40:07 2014 -0700
+++ b/mercurial/wireproto.py	Fri Mar 28 14:40:31 2014 -0700
@@ -605,6 +605,7 @@
     cg = repo.getbundle('serve', **opts)
     return streamres(proto.groupchunks(cg))
 
+@wireprotocommand('heads')
 def heads(repo, proto):
     h = repo.heads()
     return encodelist(h) + "\n"
@@ -790,7 +791,6 @@
         os.unlink(tempname)
 
 commands.update({
-    'heads': (heads, ''),
     'hello': (hello, ''),
     'known': (known, 'nodes *'),
     'listkeys': (listkeys, 'namespace'),