diff --git a/test/sampler_test.py b/test/sampler_test.py
index 1db48de2d6379d7a20a8c8d4e5ca082527a3c918..d9879543cfe8ab5c1413eb85d85b6ef806621f64 100644
--- a/test/sampler_test.py
+++ b/test/sampler_test.py
@@ -768,7 +768,7 @@ class TestPymultinest(unittest.TestCase):
                         max_modes=100, mode_tolerance=-1e90, seed=-1,
                         context=0, write_output=True, log_zero=-1e100,
                         max_iter=0, init_MPI=False, dump_callback='dumper')
-        self.sampler.kwargs['dump_callback'] = 'dumper' #Check like the dynesty print_func
+        self.sampler.kwargs['dump_callback'] = 'dumper'  # Check like the dynesty print_func
         self.assertListEqual([1, 0], self.sampler.kwargs['wrapped_params'])  # Check this separately
         self.sampler.kwargs['wrapped_params'] = None  # The dict comparison can't handle lists
         self.assertDictEqual(expected, self.sampler.kwargs)
@@ -791,7 +791,7 @@ class TestPymultinest(unittest.TestCase):
             new_kwargs[
                 "wrapped_params"
             ] = None  # The dict comparison can't handle lists
-            new_kwargs['dump_callback'] = 'dumper' #test like Dynesty print_func
+            new_kwargs['dump_callback'] = 'dumper'  # Check this like Dynesty print_func
             new_kwargs[equiv] = 123
             self.sampler.kwargs = new_kwargs
             self.assertDictEqual(expected, self.sampler.kwargs)