Fix this unit test class