diff --git a/app/models/releaf/blank_node_base.rb b/app/models/releaf/blank_node_base.rb
index <HASH>..<HASH> 100644
--- a/app/models/releaf/blank_node_base.rb
+++ b/app/models/releaf/blank_node_base.rb
@@ -16,9 +16,7 @@ module Releaf
     private
 
     def self._node_classes(klass)
-      return [klass] if klass.subclasses.blank?
-
-      classes = []
+      classes = [klass]
 
       klass.subclasses.each do |sublcass|
         classes += _node_classes(sublcass)
diff --git a/app/models/releaf/node_base.rb b/app/models/releaf/node_base.rb
index <HASH>..<HASH> 100644
--- a/app/models/releaf/node_base.rb
+++ b/app/models/releaf/node_base.rb
@@ -18,9 +18,7 @@ module Releaf
     private
 
     def self._node_classes(klass)
-      return [klass] if klass.subclasses.blank?
-
-      classes = []
+      classes = [klass]
 
       klass.subclasses.each do |sublcass|
         classes += _node_classes(sublcass)
Show all subclasses, not only bottom level