Minor change in the test