Fixed the test class.