diff --git a/google/cloud/aiplatform/tensorboard/upload_tracker.py b/google/cloud/aiplatform/tensorboard/upload_tracker.py
index fcf6da08b6..315b295e3f 100644
--- a/google/cloud/aiplatform/tensorboard/upload_tracker.py
+++ b/google/cloud/aiplatform/tensorboard/upload_tracker.py
@@ -30,10 +30,10 @@ def readable_time_string():
 
 def readable_bytes_string(bytes):
     """Get a human-readable string for number of bytes."""
-    if bytes >= 2**20:
-        return "%.1f MB" % (float(bytes) / 2**20)
-    elif bytes >= 2**10:
-        return "%.1f kB" % (float(bytes) / 2**10)
+    if bytes >= 1048576:
+        return "%.1f MB" % (bytes / 1048576)
+    elif bytes >= 1024:
+        return "%.1f kB" % (bytes / 1024)
     else:
         return "%d B" % bytes