update docs source
minor import fix