From 0a2493eca76f935dfa21119dddfde5e6560f733f Mon Sep 17 00:00:00 2001
From: MasterEnoc <brianhernandez222@hotmail.com>
Date: Fri, 27 Nov 2020 12:44:51 -0600
Subject: [PATCH] Fix 'notable traits' sections

---
 lib/docs/filters/rust/clean_html.rb | 26 ++++++++++++--------------
 1 file changed, 12 insertions(+), 14 deletions(-)

diff --git a/lib/docs/filters/rust/clean_html.rb b/lib/docs/filters/rust/clean_html.rb
index 682bf484..245ed701 100644
--- a/lib/docs/filters/rust/clean_html.rb
+++ b/lib/docs/filters/rust/clean_html.rb
@@ -27,6 +27,18 @@ module Docs
           end
         end
 
+        # Fix notable trait sections
+        css('.method, .rust.trait').each do |node|
+          traitSection = node.at_css('.notable-traits')
+
+          if traitSection
+            traitSectionContent = traitSection.css('.notable-traits-tooltiptext')
+            traitSection.css('.notable-traits-tooltip').remove
+            traitSection.add_child(traitSectionContent)
+            node.after(traitSection)
+          end
+        end
+
         css('.rusttest', '.test-arrow', 'hr').remove
 
         css('.docblock.attributes').each do |node|
@@ -88,20 +100,6 @@ module Docs
 
         css('.collapse-toggle').remove
 
-        # Fix how notable-traits sections are shown
-
-        css('.method').each do |node|
-          traitSection = node.at_css('.notable-traits')
-
-          if traitSection
-            traitSectionContent = traitSection.css('.notable-traits-tooltiptext')
-            traitSection.css('.notable-traits-tooltip').remove
-            traitSection.add_child(traitSectionContent)
-            node.after(traitSection)
-          end
-
-        end
-
         doc
       end
     end