Add missing parameter to benchmark wrapper
Add a little work to the stub.