Little fix in the test