minor change to tests