minor change in tests