Fixed import errors

Signed-off-by: Gary Smith <bk@kc.gs>