diff --git a/searchlight_ui/tests/api_tests/test_searchlight_rest.py b/searchlight_ui/tests/api_tests/test_searchlight_rest.py
index 84f65fe..9e11949 100644
--- a/searchlight_ui/tests/api_tests/test_searchlight_rest.py
+++ b/searchlight_ui/tests/api_tests/test_searchlight_rest.py
@@ -12,7 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import mock
+from unittest import mock
+
 from openstack_dashboard.test import helpers as test
 
 from searchlight_ui.api.rest import searchlight