missing imports

Signed-off-by: Marc Lefrançois <mlefrancois@mirego.com>