[AMS] fix small issue with last AMS merge