Fix scale flops2 bench bug