Fix an error with test registration