1 diff --git c/setup.py w/setup.py
2 index f87b026..6312871 100644
5 @@ -263,11 +263,6 @@ setuptools.setup(
16 os.path.join('examples', 'pylink-rtt'),
17 diff --git c/tests/unit/protocols/test_swd.py w/tests/unit/protocols/test_swd.py
18 index 03652f6..1a74455 100644
19 --- c/tests/unit/protocols/test_swd.py
20 +++ w/tests/unit/protocols/test_swd.py
22 import pylink.protocols.swd as swd
26 +from unittest import mock
30 diff --git c/tests/unit/test_decorators.py w/tests/unit/test_decorators.py
31 index ba2096b..2430cc3 100644
32 --- c/tests/unit/test_decorators.py
33 +++ w/tests/unit/test_decorators.py
36 import pylink.decorators as decorators
39 +from unittest import mock
43 diff --git c/tests/unit/test_jlink.py w/tests/unit/test_jlink.py
44 index fa7c32f..8beb2b9 100644
45 --- c/tests/unit/test_jlink.py
46 +++ w/tests/unit/test_jlink.py
47 @@ -20,7 +20,7 @@ import pylink.structs as structs
48 import pylink.unlockers.unlock_kinetis as unlock_kinetis
49 import pylink.util as util
52 +from unittest import mock
56 diff --git c/tests/unit/test_jlock.py w/tests/unit/test_jlock.py
57 index 7913b86..7492900 100644
58 --- c/tests/unit/test_jlock.py
59 +++ w/tests/unit/test_jlock.py
62 import pylink.jlock as jlock
65 +from unittest import mock
69 diff --git c/tests/unit/test_library.py w/tests/unit/test_library.py
70 index fb92f92..3b06800 100644
71 --- c/tests/unit/test_library.py
72 +++ w/tests/unit/test_library.py
73 @@ -16,7 +16,7 @@ from platform import platform
74 import pylink.library as library
75 import pylink.util as util
78 +from unittest import mock
82 diff --git c/tests/unit/test_main.py w/tests/unit/test_main.py
83 index 9b50396..c9cadbd 100644
84 --- c/tests/unit/test_main.py
85 +++ w/tests/unit/test_main.py
86 @@ -16,7 +16,7 @@ import pylink
87 import pylink.__main__ as main
91 +from unittest import mock
95 diff --git c/tests/unit/test_util.py w/tests/unit/test_util.py
96 index c60d6da..67f70a0 100644
97 --- c/tests/unit/test_util.py
98 +++ w/tests/unit/test_util.py
100 import pylink.enums as enums
101 import pylink.util as util
104 +from unittest import mock
108 diff --git c/tests/unit/unlockers/test_unlock.py w/tests/unit/unlockers/test_unlock.py
109 index e625745..ce4f0a6 100644
110 --- c/tests/unit/unlockers/test_unlock.py
111 +++ w/tests/unit/unlockers/test_unlock.py
114 import pylink.unlockers as unlock
117 +from unittest import mock
121 diff --git c/tests/unit/unlockers/test_unlock_kinetis.py w/tests/unit/unlockers/test_unlock_kinetis.py
122 index e1c0738..1e80f8d 100644
123 --- c/tests/unit/unlockers/test_unlock_kinetis.py
124 +++ w/tests/unit/unlockers/test_unlock_kinetis.py
125 @@ -16,7 +16,7 @@ import pylink.enums as enums
126 import pylink.protocols.swd as swd
127 import pylink.unlockers as unlock
130 +from unittest import mock