Fixed the OnSuccess method