Fix python3 deprecation warnings

1 files changed, 1 insertions, 5 deletions

diff --git a/tests/test_pandora/test_clientbuilder.py b/tests/test_pandora/test_clientbuilder.py
index b953f9e..db9f885 100644
--- a/tests/test_pandora/test_clientbuilder.py
+++ b/tests/test_pandora/test_clientbuilder.py
@@ -1,12 +1,8 @@
 import os
 from unittest import TestCase
 
-try:
-    from unittest.mock import Mock
-except ImportError:
-    from mock import Mock
-
 import pandora.clientbuilder as cb
+from pandora.py2compat import Mock
 from pandora import APIClient, DEFAULT_API_HOST