Fix benchmark running