small changes to the test