fixed the benchmarking.py