Tweak retainedSet

Author: odersky

compiler/src/dotty/tools/dotc/cc/CaptureOps.scala

@@ -66,8 +66,9 @@ extension (tree: Tree)
         refs
 
   /** The type representing the capture set of @retains, @retainsCap or @retainsByName annotation. */
-  def retainedSet(using Context): Type = {
-    if tree.symbol.maybeOwner == defn.RetainsCapAnnot then
+  def retainedSet(using Context): Type =
+    val rcap = defn.RetainsCapAnnot
+    if tree.symbol == rcap || tree.symbol.maybeOwner == rcap then
       defn.captureRoot.termRef
     else tree match
       case Apply(TypeApply(_, refs :: Nil), _) => refs.tpe
@@ -76,7 +77,6 @@ extension (tree: Tree)
           case AppliedType(_, refs :: Nil) => refs
           case _ => NoType
       case _ => NoType
-  }//.showing(i"retained from $tree = $result")
 
 extension (tp: Type)