private void highlightKeyword(@NotNull PsiElement node, @NotNull PyElementType elementType) {
    highlightAsKeyword(node.getNode().findChildByType(elementType));
  }
private void highlightKeyword(@NotNull PsiElement node, @NotNull PyElementType elementType)