Fix to work with wrk as well