Wrong import in documentation