From 2f8e5403cd21d18e14eedf124bd2270516dcac61 Mon Sep 17 00:00:00 2001
From: Sam Thursfield <sam.thursfield@codethink.co.uk>
Date: Wed, 19 Feb 2014 11:37:18 +0000
Subject: Fix silliness

---
 test_trove_upgrades.py | 2 +-
 util.py                | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

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
 
-- 
cgit v1.2.1