fix doc (missing import)