Add missing import

Signed-off-by: Stéphane Graber <stgraber@ubuntu.com>