minor tweak of lgm_bench.py