Adjusted the test class.