small import fix