A small change to the test class