Fix to API change
Signed-off-by: Sho Nakatani <lay.sakura@gmail.com>