Change delegate method to reflect new name of class