Fix silliness

2 files changed, 2 insertions, 1 deletions

diff --git a/test_trove_upgrades.py b/test_trove_upgrades.py
index b564709..a6d940b 100644
--- a/test_trove_upgrades.py
+++ b/test_trove_upgrades.py
@@ -331,7 +331,7 @@ class SystemTestBranch(object):
         Uses `morph edit` and `git reset` to achieve this.
         '''
         chunk_repo_uri, original_ref = self.get_repo_and_ref_for_chunk(
-                branch, stratum, chunk)
+                stratum, chunk)
 
         with set_directory(self.morphs_dir):
             run_morph(['edit', system, stratum, chunk])
diff --git a/util.py b/util.py
index 8974674..15c9c72 100644
--- a/util.py
+++ b/util.py
@@ -17,6 +17,7 @@
 
 import cliapp
 import contextlib
+import os
 import sys
 import urlparse