Updated for delegate method added

Updated for the new delegate method added.