lib/power_stencil/project/completion.rb in power_stencil-0.8.11 vs lib/power_stencil/project/completion.rb in power_stencil-0.8.12

@@ -4,20 +4,18 @@
     module Completion
 
       def query_for_completion(query_type)
         case query_type
         when :entities
-          engine.entities(engine.root_universe).map(&:as_path).sort
+          engine.entities.map(&:as_path).sort
         when :'entity-types'
           engine.available_entity_types.sort
         when :scenario
-          engine.entities(engine.root_universe, criterion: :by_type, value: :entity_override) do |entity|
+          engine.entities(criterion: :by_type, value: :entity_override) do |entity|
             !entity.scenario.nil? and !entity.scenario.empty?
           end.map(&:scenario).sort.uniq
         when :buildable
-          engine.entities(engine.root_universe) do |entity|
-            entity.buildable?
-          end.map(&:as_path).sort
+          engine.entities(&:buildable?).map(&:as_path).sort
         else
           raise PowerStencil::Error, "'#{query_type}' is not a valid query type for completion !"
         end
       end