X-Git-Url: https://code.grnet.gr/git/snf-image-creator/blobdiff_plain/8eea55728ebcad21fa0d555229ac89e72e53dab8..a939e3b86c906d078be60d8addf60865765804c0:/image_creator/disk.py

diff --git a/image_creator/disk.py b/image_creator/disk.py
index d2c4458..3e59830 100644
--- a/image_creator/disk.py
+++ b/image_creator/disk.py
@@ -34,8 +34,7 @@
 from image_creator.util import get_command
 from image_creator.util import FatalError
 from image_creator.gpt import GPTPartitionTable
-
-import image_creator.bundle_volume
+from image_creator.bundle_volume import BundleVolume
 
 import stat
 import os
@@ -101,7 +100,7 @@ class Disk(object):
     def _dir_to_disk(self):
         if self.source == '/':
             bundle = BundleVolume(self.out, self.meta)
-            return _losetup(bundle.create_image())
+            return self._losetup(bundle.create_image())
         raise FatalError("Using a directory as media source is supported")
 
     def cleanup(self):