1diff --git a/tests/dataflow_test.py b/tests/dataflow_test.py
2index e9ac3cdb..edef8200 100644
3--- a/tests/dataflow_test.py
4+++ b/tests/dataflow_test.py
5@@ -5,7 +5,7 @@ import sys
6 sys.path.append('.')
7
8 import collections
9-import mock
10+from unittest import mock
11 import unittest
12 from androguard.decompiler.dad import dataflow
13 from androguard.decompiler.dad import graph
14diff --git a/tests/test_types.py b/tests/test_types.py
15index 127dfc20..f1c89f07 100644
16--- a/tests/test_types.py
17+++ b/tests/test_types.py
18@@ -182,7 +182,7 @@ class TypesTest(unittest.TestCase):
19 for i in filter(lambda x: 'const' in x.get_name(), method.get_instructions()):
20 i.show(0)
21 # ins should only have one literal
22- self.assertEquals(len(i.get_literals()), 1)
23+ self.assertEqual(len(i.get_literals()), 1)
24
25 fmt, value = VALUES[method.full_name].pop(0)
26 converted = format_value(i.get_literals()[0], i, fmt)