Update to use the new method signatures.

Signed-off-by: Blaž Hrastnik <blaz.hrast@gmail.com>