small correction to test class