diff --git a/docs/docs/integrations/llms/rwkv.mdx b/docs/docs/integrations/llms/rwkv.mdx
index b3bfc76038cf6..7b3dda3eb6893 100644
--- a/docs/docs/integrations/llms/rwkv.mdx
+++ b/docs/docs/integrations/llms/rwkv.mdx
@@ -34,14 +34,13 @@ including streaming and CUDA support.
 
 ## Usage
 
-### RWKV
 
 To use the RWKV wrapper, you need to provide the path to the pre-trained model file and the tokenizer's configuration.
 ```python
 from langchain_community.llms import RWKV
 ```
 
-# Test the model
+## Test the model
 
 ```python
 
@@ -66,7 +65,6 @@ def generate_prompt(instruction, input=None):
 # Response:
 """
 
-
 model = RWKV(model="./models/RWKV-4-Raven-3B-v7-Eng-20230404-ctx4096.pth", strategy="cpu fp32", tokens_path="./rwkv/20B_tokenizer.json")
 response = model.invoke(generate_prompt("Once upon a time, "))
 ```