diff --git a/httpgo.go b/httpgo.go
index 67be524..5f90d95 100644
--- a/httpgo.go
+++ b/httpgo.go
@@ -65,6 +65,12 @@ func genRecords(size string) ([]Record, error) {
 	if err != nil {
 		return nil, err
 	}
+	if suffix == "MB" {
+		total *= 1024
+	}
+	if suffix == "BB" {
+		total *= 1024 * 1024
+	}
 	records = genNRecords(total)
 	return records, nil
 }