Address the merge request review comments.

4 jobs for improve-memory-benchmark in 10 minutes and 28 seconds (queued for 6 minutes and 46 seconds)