make test class smaller