Update docs with the right import statement