minor bug with test class