Update callback code to work with AR 4