--- a/hgext/rebase.py	Wed Apr 29 09:30:28 2009 +0200
+++ b/hgext/rebase.py	Fri May 01 12:26:37 2009 +0200
@@ -172,6 +172,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