fixed import in documentation