diff --git a/xos/tosca/flavorselect.py b/xos/tosca/flavorselect.py
index 36024f9..2f40bcf 100644
--- a/xos/tosca/flavorselect.py
+++ b/xos/tosca/flavorselect.py
@@ -18,7 +18,11 @@
         return int(s)
 
     def get_mb(self, s):
-        return self.get_gb(s) * 1024
+        if "GB" in s:
+            return int(s.split("GB")[0].strip())*1024
+        if "MB" in s:
+            return int(s.split("MB")[0].strip())
+        return int(s)
 
     def get_flavor(self):
         flavor = "m1.tiny"