Cleaning and documenting cortexlabdriver.

[sfa.git] / sfa / cortexlab / cortexlabshell.py

diff --git a/sfa/cortexlab/cortexlabshell.py b/sfa/cortexlab/cortexlabshell.py
index d0f5834..d68f635 100644 (file)
--- a/sfa/cortexlab/cortexlabshell.py
+++ b/sfa/cortexlab/cortexlabshell.py
@@ -21,7 +21,8 @@ from sfa.trust.certificate import Keypair, convert_public_key
 from sfa.trust.gid import create_uuid
 from sfa.trust.hierarchy import Hierarchy
 
-from sfa.cortexlab.iotlabaggregate import iotlab_xrn_object
+from sfa.iotlab.iotlabaggregate import iotlab_xrn_object
+from sfa.cortexlab.cortexlabnodes import CortexlabQueryTestbed
 
 class CortexlabShell():
     """ Class enabled to use LDAP and OAR api calls. """
@@ -38,7 +39,7 @@ class CortexlabShell():
         :type config: Config object
         """
         self.leases_db = TestbedAdditionalSfaDB(config)
-        self.query_sites = CortexlabQueryNodes()
+        self.query_sites = CortexlabQueryTestbed()
         self.ldap = LDAPapi()
         self.time_format = "%Y-%m-%d %H:%M:%S"
         self.root_auth = config.SFA_REGISTRY_ROOT_AUTH