diff --git a/merico/docstring/command.yml b/merico/docstring/command.yml
index b4f665c..1b8c71a 100644
--- a/merico/docstring/command.yml
+++ b/merico/docstring/command.yml
@@ -1,3 +1,4 @@
 description: Automatically add docstrings. Select a function or method and execute this command to generate docstring.
+help: README.md
 steps:
   - run: $devchat_python $command_path/main.py
diff --git a/merico/docstring/main.py b/merico/docstring/main.py
index 33fd1c3..e5ac3e7 100644
--- a/merico/docstring/main.py
+++ b/merico/docstring/main.py
@@ -2,10 +2,10 @@ import os
 import re
 import sys
 
-from devchat.ide.service import IDEService
-from devchat.ide.vscode_services import selected_lines, visible_lines
 from devchat.llm import chat
 
+from lib.ide_service import IDEService
+
 
 def get_selected_code():
     """