lib/hieracles/formats/yaml.rb in hieracles-0.1.5 vs lib/hieracles/formats/yaml.rb in hieracles-0.1.6

@@ -89,10 +89,15 @@
       private
 
       def added(output, key, leaf, params)
         output += leaf.to_s
         if params["#{key.join('.')}"]
-          output += " # " + params[key.join('.')][0][:file]
+          params["#{key.join('.')}"].each do |k|
+            if k[:value].to_s == leaf.to_s
+              output += " # " + k[:file]
+              break
+            end
+          end
         end
         output
       end
 
     end