Fix AAM bugs