diff --git a/python-kafka-python.changes b/python-kafka-python.changes
index 7666fde..82abca8 100644
--- a/python-kafka-python.changes
+++ b/python-kafka-python.changes
@@ -1,5 +1,5 @@
 -------------------------------------------------------------------
-Tue Oct 11 06:50:00 UTC 2022 - Steve Kowalik <steven.kowalik@suse.com>
+Wed Oct 12 02:26:13 UTC 2022 - Steve Kowalik <steven.kowalik@suse.com>
 
 - Add patch remove-mock.patch:
   * Remove use of mock. 
diff --git a/remove-mock.patch b/remove-mock.patch
index 0372079..868c35f 100644
--- a/remove-mock.patch
+++ b/remove-mock.patch
@@ -14,3 +14,19 @@ Index: kafka-python-2.0.2/test/test_consumer_integration.py
  import pytest
  from kafka.vendor.six.moves import range
  
+Index: kafka-python-2.0.2/test/test_conn.py
+===================================================================
+--- kafka-python-2.0.2.orig/test/test_conn.py
++++ kafka-python-2.0.2/test/test_conn.py
+@@ -4,7 +4,10 @@ from __future__ import absolute_import
+ from errno import EALREADY, EINPROGRESS, EISCONN, ECONNRESET
+ import socket
+ 
+-import mock
++try:
++    from unittest import mock
++except ImportError:
++    import mock
+ import pytest
+ 
+ from kafka.conn import BrokerConnection, ConnectionStates, collect_hosts