Fix methods to match the current Python API

Signed-off-by: Claudio Matsuoka <cmatsuoka@gmail.com>