A small change in the unit test class