Fix exception due to old method name