Yes, the facts as presented allow a 4th year of AOTC. Since the student is still an undergrad on 1/1/22 both the undergrad expenses and grad school expenses count for 2022. It's a quirky rule.
