Added remaining missing AM callbacks.

Change-Id: I51c77c716147b82effa263f56e921f6a71f2d5f5