diff --git a/test/unit/test_providers_raw_master_key_provider.py b/test/unit/test_providers_raw_master_key_provider.py
index 568955bcd..5128b1e22 100644
--- a/test/unit/test_providers_raw_master_key_provider.py
+++ b/test/unit/test_providers_raw_master_key_provider.py
@@ -11,11 +11,8 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 """Test suite for aws_encryption_sdk.key_providers.raw.RawMasterKeyProvider"""
-import unittest
-
 import attr
 import pytest
-import six
 from mock import MagicMock, patch, sentinel
 
 from aws_encryption_sdk.key_providers.base import MasterKeyProvider, MasterKeyProviderConfig
@@ -42,7 +39,7 @@ def _get_raw_key(self, key_id):
         return self.config.mock_wrapping_key
 
 
-class TestRawMasterKeyProvider(unittest.TestCase):
+class TestRawMasterKeyProvider(object):
     def test_parent(self):
         assert issubclass(RawMasterKeyProvider, MasterKeyProvider)
 
@@ -50,8 +47,9 @@ def test_get_raw_key_enforcement(self):
         class TestProvider(RawMasterKeyProvider):
             pass
 
-        with six.assertRaisesRegex(self, TypeError, "Can't instantiate abstract class TestProvider *"):
+        with pytest.raises(TypeError) as excinfo:
             TestProvider()
+        excinfo.match("Can't instantiate abstract class TestProvider *")
 
     @patch(
         "aws_encryption_sdk.key_providers.raw.RawMasterKeyConfig", return_value=sentinel.raw_master_key_config_instance