fix usage for mock