Add test for branching off not from master

1 files changed, 2 insertions, 2 deletions

diff --git a/morph b/morph
index 8f3ba311..0a77995b 100755
--- a/morph
+++ b/morph
@@ -719,13 +719,13 @@ class Morph(cliapp.Application):
     def cmd_branch(self, args):
         '''Branch the whole system.'''
         
-        if len(args) != 1:
+        if len(args) not in [1, 2]:
             raise cliapp.AppException('morph branch needs name of branch '
                                         'as parameter')
 
         new_branch = args[0]
         repo = 'morphs'
-        commit = 'master'
+        commit = 'master' if len(args) == 1 else args[1]
 
         # Create the system branch directory.
         os.makedirs(new_branch)