Removed unnecessary class

Removed unnecessary class from the test case.