--- a/hgext/rebase.py	Thu Apr 30 23:57:36 2009 -0500
+++ b/hgext/rebase.py	Fri May 01 12:35:13 2009 +0200
@@ -176,6 +176,7 @@
                             user=repo[rev].user(),
                             date=repo[rev].date(),
                             extra=extra)
+        repo.dirstate.setbranch(repo[newrev].branch())
         return newrev
     except util.Abort:
         # Invalidate the previous setparents