fix import

Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>