Use correct module name in doc string