Fix case-related augment test