diff --git a/python/helpers/pycharm/docrunner.py b/python/helpers/pycharm/docrunner.py
index 1a6c3cc64053..baa28f002c6c 100644
--- a/python/helpers/pycharm/docrunner.py
+++ b/python/helpers/pycharm/docrunner.py
@@ -180,11 +180,7 @@ class DocTestRunner(doctest.DocTestRunner):
         self.result.startTest(example)
         err = self._failure_header(test, example) +\
               self._checker.output_difference(example, got, self.optionflags)
-        expected = None
-        try:
-          expected = example.want
-        except KeyError:
-          pass
+        expected = getattr(example, "want", None)
         self.result.addFailure(example, err, expected=expected, actual=got)
 
       elif outcome is BOOM: