Fix an error in test