Fix documentation import statement