use selected branch as base when creating a new branch

2 files changed, 4 insertions, 4 deletions

diff --git a/pkg/commands/git.go b/pkg/commands/git.go
index 588cec80e..00ec296de 100644
--- a/pkg/commands/git.go
+++ b/pkg/commands/git.go
@@ -327,8 +327,8 @@ func (c *GitCommand) ResetToCommit(sha string, strength string) error {
 }
 
 // NewBranch create new branch
-func (c *GitCommand) NewBranch(name string) error {
-	return c.OSCommand.RunCommand("git checkout -b %s", name)
+func (c *GitCommand) NewBranch(name string, baseBranch string) error {
+	return c.OSCommand.RunCommand("git checkout -b %s %s", name, baseBranch)
 }
 
 // CurrentBranchName is a function.
diff --git a/pkg/commands/git_test.go b/pkg/commands/git_test.go
index fabeb6375..779e31f68 100644
--- a/pkg/commands/git_test.go
+++ b/pkg/commands/git_test.go
@@ -636,12 +636,12 @@ func TestGitCommandNewBranch(t *testing.T) {
 	gitCmd := NewDummyGitCommand()
 	gitCmd.OSCommand.command = func(cmd string, args ...string) *exec.Cmd {
 		assert.EqualValues(t, "git", cmd)
-		assert.EqualValues(t, []string{"checkout", "-b", "test"}, args)
+		assert.EqualValues(t, []string{"checkout", "-b", "test", "master"}, args)
 
 		return exec.Command("echo")
 	}
 
-	assert.NoError(t, gitCmd.NewBranch("test"))
+	assert.NoError(t, gitCmd.NewBranch("test", "master"))
 }
 
 // TestGitCommandDeleteBranch is a function.