diff --git a/tests/core/host/test_autonat.py b/tests/core/host/test_autonat.py
index fe394745..66fde55b 100644
--- a/tests/core/host/test_autonat.py
+++ b/tests/core/host/test_autonat.py
@@ -14,7 +14,7 @@ from libp2p.host.autonat.pb.autonat_pb2 import (
     DialRequest,
     DialResponse,
     Message,
-    PeerInfo,
+    Peer_Info,
     Status,
     Type,
 )
@@ -137,7 +137,7 @@ async def test_handle_dial():
         # Create a mock message with a peer to dial
         message = Message()
         message.type = Type.Value("DIAL")
-        peer_info = PeerInfo()
+        peer_info = Peer_Info()
         peer_info.id = peer_id.to_bytes()
         peer_info.addrs.extend([b"/ip4/127.0.0.1/tcp/4001"])
         message.dial.peers.append(peer_info)
@@ -169,7 +169,7 @@ async def test_handle_request():
         message = Message()
         message.type = Type.DIAL
         dial_request = DialRequest()
-        peer_info = PeerInfo()
+        peer_info = Peer_Info()
         dial_request.peers.append(peer_info)
         message.dial.CopyFrom(dial_request)
 
@@ -208,7 +208,7 @@ async def test_handle_stream():
         request = Message()
         request.type = Type.DIAL
         dial_request = DialRequest()
-        peer_info = PeerInfo()
+        peer_info = Peer_Info()
         peer_info.id = b"peer_id"
         peer_info.addrs.append(b"addr1")
         dial_request.peers.append(peer_info)
diff --git a/tox.ini b/tox.ini
index 983c3ee4..0528f198 100644
--- a/tox.ini
+++ b/tox.ini
@@ -73,3 +73,8 @@ commands=
     bash.exe -c 'python -m pip install --upgrade "$(ls dist/libp2p-*-py3-none-any.whl)" --progress-bar off'
     python -c "import libp2p"
 skip_install=true
+
+[testenv:docs]
+extras=.
+commands =
+    make check-docs-ci