Fix the variant choice type