Updating the test failure class

Updating the test failure class