| diff --git a/test/py/cmdlib/test_unittest.py b/test/py/cmdlib/test_unittest.py |
| index f93f99d..2d0d083 100644 |
| --- a/test/py/cmdlib/test_unittest.py |
| +++ b/test/py/cmdlib/test_unittest.py |
| @@ -85,7 +85,7 @@ class TestLUTestDelay(CmdlibTestCase): |
| |
| self.ExecOpCode(op) |
| |
| - self.rpc.call_test_delay.assert_called_once() |
| + assert self.rpc.call_test_delay.called |
| |
| def testFailingRpc(self): |
| op = opcodes.OpTestDelay(duration=DELAY_DURATION, |
| diff --git a/test/py/testutils/__init__.py b/test/py/testutils/__init__.py |
| index 27ca425..04ea0ee 100644 |
| --- a/test/py/testutils/__init__.py |
| +++ b/test/py/testutils/__init__.py |
| @@ -242,7 +242,7 @@ def patch_object(*args, **kwargs): |
| This function unifies the different variations. |
| |
| """ |
| - import mock |
| + from mock import mock |
| try: |
| # pylint: disable=W0212 |
| return mock._patch_object(*args, **kwargs) |