Fix hg code: now displays correct branch name