templates/default/layout/dot/setup.rb in yard-0.4.0 vs templates/default/layout/dot/setup.rb in yard-0.5.0

@@ -1,9 +1,9 @@
 attr_reader :contents
 
 def init
   if object
-    type = object == Registry.root ? :module : object.type
+    type = object.root? ? :module : object.type
     sections :header, [T(type)]
   else
     sections :header, [:contents]
   end
 end