--- a/tests.py
+++ b/tests.py
@@ -7,7 +7,7 @@
 import unittest
 import time
 import re
-import mock
+from unittest import mock
 from contextlib import contextmanager
 
 from email.header import Header
@@ -82,7 +82,7 @@
             self.app.testing
         )
 
-        self.assertEquals(self.mail.state.__dict__, mail.__dict__)
+        self.assertEqual(self.mail.state.__dict__, mail.__dict__)
 
 
 class TestMessage(TestCase):