diff --git a/lib/retriever/fetchsitemap.rb b/lib/retriever/fetchsitemap.rb
index 725857b..71c92b6 100644
--- a/lib/retriever/fetchsitemap.rb
+++ b/lib/retriever/fetchsitemap.rb
@@ -16,8 +16,6 @@ def initialize(url, options)
       @result.uniq!
     end
 
-    private
-
     # produces valid XML sitemap based on page collection fetched.
     # Writes to current directory.
     def gen_xml
@@ -33,6 +31,8 @@ def gen_xml
       print_file_info(filename)
     end
 
+    private
+
     def print_file_info(filename)
       puts HR
       puts "File Created: sitemap-#{filename}.xml"
diff --git a/spec/retriever_spec.rb b/spec/retriever_spec.rb
index de9bf04..a4cdcfd 100644
--- a/spec/retriever_spec.rb
+++ b/spec/retriever_spec.rb
@@ -66,4 +66,25 @@
       expect(success_resp).to eq(true)
     end
   end
+
+  describe Retriever::FetchSitemap do
+    let(:options) do
+      { limit: 1, progress: false }
+    end
+
+    let(:url) { 'http://www.yahoo.com' }
+
+    let(:removed_sitemap) { FileUtils.rm(Dir.pwd + '/sitemap-yahoo.xml').first }
+
+    subject { described_class.new(url, options) }
+
+    before do
+      subject.gen_xml
+    end
+
+    it 'generates xml' do
+      sitemap_file = removed_sitemap
+      expect(sitemap_file).to match 'sitemap-yahoo.xml'
+    end
+  end
 end