fix doc
Better module import