Use callback to fix ai animation inconsistency