SDK upgrade callback fix