Migrate `Message_shift` calls to new API