Add condition to tests