Made changes to failed test method