Fix imports

Signed-off-by: Jorge Ruesga <jorge@ruesga.com>