Add 2 missing cases