Fix method calls to reflect API changes