More fixes to amb handling