diff --git a/wal/wal_bench_test.go b/wal/wal_bench_test.go
index f27f1af8e..9568399b2 100644
--- a/wal/wal_bench_test.go
+++ b/wal/wal_bench_test.go
@@ -53,6 +53,7 @@ func benchmarkWriteEntry(b *testing.B, size int, batch int) {
 
 	b.ResetTimer()
 	n := 0
+	b.SetBytes(int64(e.Size()))
 	for i := 0; i < b.N; i++ {
 		err := w.saveEntry(e)
 		if err != nil {