diff --git a/lib/vmfloaty.rb b/lib/vmfloaty.rb
index 2965b83..4ff3e2b 100644
--- a/lib/vmfloaty.rb
+++ b/lib/vmfloaty.rb
@@ -64,7 +64,7 @@ class Vmfloaty
 
         unless os_types.nil?
           response = Pooler.retrieve(verbose, os_types, token, url)
-          Format.get_hosts(response)
+          puts Format.get_hosts(response)
         else
           puts 'You did not provide an OS to get'
         end
diff --git a/lib/vmfloaty/format.rb b/lib/vmfloaty/format.rb
index fb263a4..d89e700 100644
--- a/lib/vmfloaty/format.rb
+++ b/lib/vmfloaty/format.rb
@@ -14,6 +14,6 @@ class Format
       end
     end
 
-    puts host_hash.to_json
+    host_hash.to_json
   end
 end
diff --git a/spec/vmfloaty/format_spec.rb b/spec/vmfloaty/format_spec.rb
new file mode 100644
index 0000000..9442225
--- /dev/null
+++ b/spec/vmfloaty/format_spec.rb
@@ -0,0 +1,18 @@
+require 'spec_helper'
+require 'json'
+require_relative '../../lib/vmfloaty/format'
+
+describe Pooler do
+
+  describe "#get_hosts" do
+    before :each do
+      @hostname_hash = "{\"ok\":true,\"debian-7-i386\":{\"hostname\":[\"sc0o4xqtodlul5w\",\"4m4dkhqiufnjmxy\"]},\"debian-7-x86_64\":{\"hostname\":\"zb91y9qbrbf6d3q\"},\"domain\":\"company.com\"}"
+      @format_hash = "{\"debian-7-i386\":[\"sc0o4xqtodlul5w\",\"4m4dkhqiufnjmxy\"],\"debian-7-x86_64\":\"zb91y9qbrbf6d3q\",\"domain\":\"company.com\"}"
+    end
+
+    it "formats a hostname hash into os, hostnames, and domain name" do
+
+      expect(Format.get_hosts(JSON.parse(@hostname_hash))).to eq @format_hash
+    end
+  end
+end