Removed unused class from unit test