3 files changed, 9 insertions, 9 deletions
diff --git a/rumba/testbeds/emulab.py b/rumba/testbeds/emulab.py
index b7c0c51..cbaa730 100644
--- a/rumba/testbeds/emulab.py
+++ b/rumba/testbeds/emulab.py
@@ -25,7 +25,7 @@ import re
 from ast import literal_eval
 
 import rumba.ssh_support as ssh
-from rumba.model import Testbed
+import rumba.model as mod
 
 import warnings
 warnings.filterwarnings("ignore")
@@ -35,11 +35,11 @@ warnings.filterwarnings("ignore")
 # @url [string] URL of the testbed
 # @image [string] specific image to use
 #
-class EmulabTestbed(Testbed):
+class Testbed(mod.Testbed):
     def __init__(self, exp_name, username, password = "",
                  proj_name  = "ARCFIRE", url = "wall1.ilabt.iminds.be",
                  image = "UBUNTU14-64-STD"):
-        Testbed.__init__(self, exp_name, username, password, proj_name)
+        mod.Testbed.__init__(self, exp_name, username, password, proj_name)
         self.url = url
         self.image = image
 
diff --git a/rumba/testbeds/faketestbed.py b/rumba/testbeds/faketestbed.py
index f163411..e47eb85 100644
--- a/rumba/testbeds/faketestbed.py
+++ b/rumba/testbeds/faketestbed.py
@@ -18,13 +18,13 @@
 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 # MA  02110-1301  USA
 
-from rumba.model import Testbed
+import rumba.model as mod
 
 # Fake testbed, useful for testing
-class FakeTestbed(Testbed):
+class Testbed(mod.Testbed):
     def __init__(self, exp_name, username, proj_name = "ARCFIRE",
                  password = ""):
-        Testbed.__init__(self, exp_name, username, password, proj_name)
+        mod.Testbed.__init__(self, exp_name, username, password, proj_name)
 
     def create_experiment(self, nodes, links):
         print("[Fake testbed] experiment swapped in")
diff --git a/rumba/testbeds/jfed.py b/rumba/testbeds/jfed.py
index 5f4d496..92e615f 100644
--- a/rumba/testbeds/jfed.py
+++ b/rumba/testbeds/jfed.py
@@ -22,13 +22,13 @@ import subprocess
 import getpass
 import xml.dom.minidom as xml
 
-from rumba.model import Testbed
+import rumba.model as mod
 
-class jFedTestbed(Testbed):
+class Testbed(mod.Testbed):
     def __init__(self, exp_name, username, cert_file, jfed_jar, exp_hours = "2",
                  proj_name = "ARCFIRE", authority = "wall2.ilabt.iminds.be"):
         passwd = getpass.getpass(prompt = "Password for certificate file: ")
-        Testbed.__init__(self, exp_name, username, passwd, proj_name)
+        mod.Testbed.__init__(self, exp_name, username, passwd, proj_name)
         self.authority = "urn:publicid:IDN+" + authority + "+authority+cm"
         self.auth_name = authority
         self.cert_file = cert_file