For test class