fix import

Signed-off-by: Gaetan Semet <gaetan@xeberon.net>