The API has changed here (due to compatibility with Python 3), so once
new API is available we have to use it.
Signed-off-by: Michal Čihař <michal@cihar.com>