Fixed error in test class.