Fix an error in the test