Add parameter to tests