Refactor disk class

[nitsi.git] / src / nitsi / machine.py

diff --git a/src/nitsi/machine.py b/src/nitsi/machine.py
index 057981db62098e11b45e7c8d6dc577e55f80c008..9aed5a0180dda8019d606e0c5be10536291bec7d 100644 (file)
--- a/src/nitsi/machine.py
+++ b/src/nitsi/machine.py
@@ -176,10 +176,11 @@ class Machine():
 
     def copy_in(self, fr, to):
         try:
-            self.disk.mount(self.root_uid, "/")
+            self.disk.inspect()
+            self.disk.mount()
             self.disk.copy_in(fr, to)
         except BaseException as e:
             self.log.error(e)
         finally:
-            self.disk.umount("/")
+            self.disk.umount()
             self.disk.close()