make import_module return the module as well

Signed-off-by: Alfredo Deza <alfredo@deza.pe>

diff --git a/remoto/connection.py b/remoto/connection.py
index a4ad30bf0b3a689ab4b029e45e7b32996d04d48e..13550c5f7cccb75ea4f0822941e6fd2e0cb4fa8a 100644 (file)
--- a/remoto/connection.py
+++ b/remoto/connection.py
@@ -76,6 +76,7 @@ class Connection(object):
 
     def import_module(self, module):
         self.remote_module = ModuleExecute(self.gateway, module, self.logger)
+        return self.remote_module
 
 
 class ModuleExecute(object):