Change to test class