Fix method name
Migrate to new API