Fix a small issue
benchmark