X-Git-Url: https://code.grnet.gr/git/snf-image-creator/blobdiff_plain/f99fe99d6f3beaf5b282267f30c1ca288be92e40..aeb95900f10211f2d3c8ffead154a44121a2d5b5:/image_creator/main.py

diff --git a/image_creator/main.py b/image_creator/main.py
index 49b76e7..85c9272 100644
--- a/image_creator/main.py
+++ b/image_creator/main.py
@@ -36,7 +36,7 @@
 from image_creator import __version__ as version
 from image_creator import util
 from image_creator.disk import Disk
-from image_creator.util import get_command, FatalError, MD5
+from image_creator.util import FatalError, MD5
 from image_creator.output.cli import SilentOutput, SimpleOutput, \
     OutputWthProgress
 from image_creator.os_type import os_cls
@@ -219,7 +219,7 @@ def image_creator():
         metadata = image_os.meta
         dev.umount()
 
-        size = options.shrink and dev.shrink() or dev.meta['SIZE']
+        size = options.shrink and dev.shrink() or dev.size
         metadata.update(dev.meta)
 
         # Add command line metadata to the collected ones...