diff --git a/topstat.py b/topstat.py
index c448e03..1023576 100644
--- a/topstat.py
+++ b/topstat.py
@@ -59,7 +59,8 @@ def convert_to_megabytes(data_size):
 
     breakpoint()
     try:
-        size, unit = data_size.upper().split()
+        unit = data_size[-1].upper()
+        size = data_size[:-1]
         size = float(size)
     except (ValueError, AttributeError):
         return None