minor import fix