you'd need to add the parameter to the Mock class too