Fix test set (Type mismatch)