diff --git a/distarray/tests/test_plotting.py b/distarray/tests/test_plotting.py
index f717ca1f..9f16a3d7 100644
--- a/distarray/tests/test_plotting.py
+++ b/distarray/tests/test_plotting.py
@@ -18,11 +18,12 @@
 
 import unittest
 
-from distarray.dist import Context, Distribution
+from distarray.testing import ContextTestCase
+from distarray.dist import Distribution
 from distarray.testing import import_or_skip
 
 
-class TestContext(unittest.TestCase):
+class TestPlotting(ContextTestCase):
     """Test Context methods"""
 
     @classmethod
@@ -30,15 +31,10 @@ def setUpClass(cls):
         # raise a skipTest if plotting import fails
         # (because matplotlib isn't installed, probably)
         cls.plt = import_or_skip("distarray.plotting")
-        cls.context = Context()
+        super(TestPlotting, cls).setUpClass()
         cls.da = Distribution.from_shape(cls.context, (64, 64))
         cls.arr = cls.context.ones(cls.da)
 
-    @classmethod
-    def tearDownClass(cls):
-        """Close the client connections"""
-        cls.context.close()
-
     def test_plot_array_distribution(self):
         # Only tests that this runs, not that it's correct
         process_coords = [(0, 0), (1, 0), (2, 0), (3, 0)]