diff --git a/internal/stats/bandwidth_stat_manager_test.go b/internal/stats/bandwidth_stat_manager_test.go
index d22cd7f..b0750aa 100644
--- a/internal/stats/bandwidth_stat_manager_test.go
+++ b/internal/stats/bandwidth_stat_manager_test.go
@@ -10,22 +10,22 @@ import (
 
 func TestBandwidthStatManager_Add(t *testing.T) {
 	var manager = stats.NewBandwidthStatManager()
-	manager.Add(1, 1, 10)
-	manager.Add(1, 1, 10)
-	manager.Add(1, 1, 10)
+	manager.Add(1, 1, 10, 10)
+	manager.Add(1, 1, 10, 10)
+	manager.Add(1, 1, 10, 10)
 	time.Sleep(1 * time.Second)
-	manager.Add(1, 1, 85)
+	manager.Add(1, 1, 85, 85)
 	time.Sleep(1 * time.Second)
-	manager.Add(1, 1, 25)
-	manager.Add(1, 1, 75)
+	manager.Add(1, 1, 25, 25)
+	manager.Add(1, 1, 75, 75)
 	manager.Inspect()
 }
 
 func TestBandwidthStatManager_Loop(t *testing.T) {
 	var manager = stats.NewBandwidthStatManager()
-	manager.Add(1, 1, 10)
-	manager.Add(1, 1, 10)
-	manager.Add(1, 1, 10)
+	manager.Add(1, 1, 10, 10)
+	manager.Add(1, 1, 10, 10)
+	manager.Add(1, 1, 10, 10)
 	err := manager.Loop()
 	if err != nil {
 		t.Fatal(err)