diff --git a/README.rst b/README.rst
index c7a82a8..808f4bd 100644
--- a/README.rst
+++ b/README.rst
@@ -103,9 +103,17 @@ Testing
 -------
 If you installed DFO-GN manually, you can test your installation by running:
 
+ .. code-block:: bash
+
+    $ pytest
+
+for Python 3.7+ (need `pytest <http://pytest.org>`_)
+
  .. code-block:: bash
 
     $ python setup.py test
 
+for Python 2.7
+
 Alternatively, the `documentation <https://numericalalgorithmsgroup.github.io/dfogn/>`_ provides some simple examples of how to run DFO-GN, which are also available in the examples directory.
 
diff --git a/dfogn/tests/test_util.py b/dfogn/tests/test_util.py
index afb6fe3..9e2a121 100644
--- a/dfogn/tests/test_util.py
+++ b/dfogn/tests/test_util.py
@@ -38,7 +38,7 @@ def runTest(self):
         n = 10
         x = np.sin(np.arange(n))
         normx = np.sum(x**2)
-        self.assertAlmostEqual(normx, sumsq(x), 'Wrong answer')
+        self.assertAlmostEqual(normx, sumsq(x), msg='Wrong answer')
 
 
 class TestEval(unittest.TestCase):