diff --git a/libp2p/host/basic_host.py b/libp2p/host/basic_host.py
index 013c62b..1b10c2c 100644
--- a/libp2p/host/basic_host.py
+++ b/libp2p/host/basic_host.py
@@ -3,6 +3,7 @@ from typing import TYPE_CHECKING, List, Sequence
 
 import multiaddr
 
+from libp2p.crypto.keys import PublicKey
 from libp2p.host.defaults import get_default_protocols
 from libp2p.host.exceptions import StreamFailure
 from libp2p.network.network_interface import INetwork
diff --git a/libp2p/host/host_interface.py b/libp2p/host/host_interface.py
index 9e55796..cfba82c 100644
--- a/libp2p/host/host_interface.py
+++ b/libp2p/host/host_interface.py
@@ -3,6 +3,7 @@ from typing import Any, List, Sequence
 
 import multiaddr
 
+from libp2p.crypto.keys import PublicKey
 from libp2p.network.network_interface import INetwork
 from libp2p.network.stream.net_stream_interface import INetStream
 from libp2p.peer.id import ID
diff --git a/libp2p/identity/identify/protocol.py b/libp2p/identity/identify/protocol.py
index 0fc65ba..b42e51a 100644
--- a/libp2p/identity/identify/protocol.py
+++ b/libp2p/identity/identify/protocol.py
@@ -1,12 +1,10 @@
 import logging
-from typing import Sequence
 
 from multiaddr import Multiaddr
 
-from libp2p.crypto.keys import PublicKey
 from libp2p.host.host_interface import IHost
 from libp2p.network.stream.net_stream_interface import INetStream
-from libp2p.typing import StreamHandlerFn, TProtocol
+from libp2p.typing import StreamHandlerFn
 
 from .pb.identify_pb2 import Identify