diff --git a/lib/parser/source/comment/associator.rb b/lib/parser/source/comment/associator.rb
index 32811b5b2..b81d49597 100644
--- a/lib/parser/source/comment/associator.rb
+++ b/lib/parser/source/comment/associator.rb
@@ -95,10 +95,11 @@ def associate
       end
 
       ##
-      # Same as {associate}, but compares by identity, thus producing an unambiguous
-      # result even in presence of equal nodes.
+      # Same as {associate}, but uses `node.loc` instead of `node` as
+      # the hash key, thus producing an unambiguous result even in presence
+      # of equal nodes.
       #
-      # @return [Hash<Parser::Source::Node, Array<Parser::Source::Comment>>]
+      # @return [Hash<Parser::Source::Map, Array<Parser::Source::Comment>>]
       #
       def associate_locations
         @map_using = :location
@@ -106,11 +107,10 @@ def associate_locations
       end
 
       ##
-      # Same as {associate}, but uses `node.loc` instead of `node` as
-      # the hash key, thus producing an unambiguous result even in presence
-      # of equal nodes.
+      # Same as {associate}, but compares by identity, thus producing an unambiguous
+      # result even in presence of equal nodes.
       #
-      # @return [Hash<Parser::Source::Map, Array<Parser::Source::Comment>>]
+      # @return [Hash<Parser::Source::Node, Array<Parser::Source::Comment>>]
       #
       def associate_by_identity
         @map_using = :identity