made changes in test class.