diff --git a/python/src/com/jetbrains/python/documentation/doctest/PyDocstringTokenSetContributor.java b/python/src/com/jetbrains/python/documentation/doctest/PyDocstringTokenSetContributor.java
index 326e246ae822..7059c6af8531 100644
--- a/python/src/com/jetbrains/python/documentation/doctest/PyDocstringTokenSetContributor.java
+++ b/python/src/com/jetbrains/python/documentation/doctest/PyDocstringTokenSetContributor.java
@@ -7,13 +7,15 @@ import com.jetbrains.python.PythonTokenSetContributor;
  * User : ktisha
  */
 public class PyDocstringTokenSetContributor extends PythonTokenSetContributor {
+  public static final TokenSet DOCSTRING_REFERENCE_EXPRESSIONS = TokenSet.create(PyDocstringTokenTypes.DOC_REFERENCE);
+
   @Override
   public TokenSet getExpressionTokens() {
-    return TokenSet.orSet(super.getExpressionTokens(), TokenSet.create(PyDocstringTokenTypes.DOC_REFERENCE));
+    return DOCSTRING_REFERENCE_EXPRESSIONS;
   }
 
   @Override
   public TokenSet getReferenceExpressionTokens() {
-    return TokenSet.orSet(super.getExpressionTokens(), TokenSet.create(PyDocstringTokenTypes.DOC_REFERENCE));
+    return DOCSTRING_REFERENCE_EXPRESSIONS;
   }
 }