bin/jsonpath in jsonpath-0.3.3 vs bin/jsonpath in jsonpath-0.4.0

@@ -1,9 +1,9 @@
 #!/usr/local/env ruby
 
 require 'jsonpath'
-require 'json'
+require 'yajl'
 
 def usage
   puts "jsonpath [expression] (file|string)
 
 If you omit the second argument, it will read stdin, assuming one valid JSON object
@@ -15,10 +15,10 @@
 
 jsonpath = JsonPath.new(ARGV[0])
 case ARGV[1]
 when nil #stdin
   until STDIN.eof?
-    puts jsonpath.on(JSON.parse(STDIN.readline)).to_a.to_json
+    puts Yajl::Encoder.encode(jsonpath.on(Yajl::Parser.parse(STDIN.readline)))
   end
 when String
-  puts jsonpath.on(JSON.parse(File.exist?(ARGV[1]) ? File.read(ARGV[1]) : ARGV[1])).to_a.to_json
+  puts Yajl::Encoder.encode(jsonpath.on(Yajl::Parser.parse(File.exist?(ARGV[1]) ? File.read(ARGV[1]) : ARGV[1])))
 end
\ No newline at end of file