API fix for method call change