Fix docs imports