Few changes in test file