diff --git a/tests/libp2p/test_notify.py b/tests/libp2p/test_notify.py
index 5ebed1b..fdf7368 100644
--- a/tests/libp2p/test_notify.py
+++ b/tests/libp2p/test_notify.py
@@ -1,11 +1,3 @@
-import pytest
-
-from tests.utils import cleanup
-from libp2p import new_node
-from libp2p.network.notifee_interface import INotifee
-
-# pylint: disable=too-many-locals
-
 """
 Test Notify and Notifee by ensuring that the proper events get
 called, and that the stream passed into opened_stream is correct
@@ -13,11 +5,18 @@ called, and that the stream passed into opened_stream is correct
 Note: Listen event does not get hit because MyNotifee is passed
 into network after network has already started listening
 
-TODO: Add tests to ensure conn is the correct connection
 TODO: Add tests for closed_stream disconnected, listen_close when those
 features are implemented in swarm
 """
 
+import pytest
+
+from tests.utils import cleanup
+from libp2p import new_node
+from libp2p.network.notifee_interface import INotifee
+
+# pylint: disable=too-many-locals
+
 class MyNotifee(INotifee):
     # pylint: disable=too-many-instance-attributes, cell-var-from-loop