diff --git a/vm.py b/vm.py
index ab708de..3d534de 100755
--- a/vm.py
+++ b/vm.py
@@ -57,8 +57,10 @@ def to_mb(value):
     v = int(value.lower()[:-1])
     if value.lower()[-1] == 'g':
         return v * 1024
-    else:
+    elif value.lower()[-1] == 'm':
         return v
+    else:
+        return int(value.lower())
 
 j2 = jinja2.Environment(loader=jinja2.FileSystemLoader('templates'))
 j2.filters['ipreplace'] = replace_ip