Add missing import

Signed-off-by: Sebastian Ramacher <sebastian+dev@ramacher.at>