fix bench code