diff --git a/pymysqlreplication/tests/test_basic.py b/pymysqlreplication/tests/test_basic.py
index 74a46088..6e3f09bc 100644
--- a/pymysqlreplication/tests/test_basic.py
+++ b/pymysqlreplication/tests/test_basic.py
@@ -220,12 +220,13 @@ def test_log_pos(self):
 
         self.assertIsInstance(self.stream.fetchone(), XidEvent)
 
-        self.assertIsNone(self.stream.fetchone())
 
     def test_log_pos_handles_disconnects(self):
+        self.stream.close()
         self.stream = BinLogStreamReader(
             connection_settings=self.database,
-            resume_stream=True
+            resume_stream=False,
+            only_events = [FormatDescriptionEvent, QueryEvent, TableMapEvent, WriteRowsEvent, XidEvent]
         )
 
         query = "CREATE TABLE test (id INT  PRIMARY KEY AUTO_INCREMENT, data VARCHAR (50) NOT NULL)"
@@ -234,25 +235,18 @@ def test_log_pos_handles_disconnects(self):
         self.execute(query)
         self.execute("COMMIT")
 
-        self.assertIsInstance(self.stream.fetchone(), RotateEvent)
-
         self.assertIsInstance(self.stream.fetchone(), FormatDescriptionEvent)
         self.assertGreater(self.stream.log_pos, 0)
-
         self.assertIsInstance(self.stream.fetchone(), QueryEvent)
+
         self.assertIsInstance(self.stream.fetchone(), QueryEvent)
         self.assertIsInstance(self.stream.fetchone(), TableMapEvent)
         self.assertIsInstance(self.stream.fetchone(), WriteRowsEvent)
-        self.assertIsInstance(self.stream.fetchone(), XidEvent)
 
-        self.assertIsNone(self.stream.fetchone())
-        self.assertIsInstance(self.stream.fetchone(), RotateEvent)
+        self.assertIsInstance(self.stream.fetchone(), XidEvent)
 
-        self.assertIsInstance(self.stream.fetchone(), FormatDescriptionEvent)
         self.assertGreater(self.stream.log_pos, 0)
 
-        #self.assertFalse(self.stream._BinLogStreamReader__connected_stream)
-
 
 class TestMultipleRowBinLogStreamReader(base.PyMySQLReplicationTestCase):
     def test_insert_multiple_row_event(self):