1 files changed, 3 insertions, 1 deletions
diff --git a/becommands/set_root.py b/becommands/set_root.py
index ad20b8e..14286e0 100644
--- a/becommands/set_root.py
+++ b/becommands/set_root.py
@@ -20,6 +20,7 @@ from libbe import bugdir, cmdutil, rcs
 def execute(args):
     """
     >>> from libbe import tests
+    >>> import os
     >>> dir = tests.Dir()
     >>> try:
     ...     bugdir.tree_root(dir.name)
@@ -32,7 +33,8 @@ def execute(args):
     >>> bd = bugdir.tree_root(dir.name)
     >>> bd.root = dir.name
     >>> dir = tests.arch_dir()
-    >>> execute([dir.name])
+    >>> os.chdir(dir.name)
+    >>> execute('.')
     Using Arch for revision control.
     Directory initialized.
     >>> bd = bugdir.tree_root(dir.name+"/{arch}")