lib/snapshot/reports_generator.rb in snapshot-0.2.1 vs lib/snapshot/reports_generator.rb in snapshot-0.2.2

@@ -35,10 +35,14 @@
       Helper.log.info "Successfully created HTML file with all the screenshots: #{export_path}".green
     end
 
     private
       def lib_path
-        "./lib/"
+        if not Helper.is_test? and Gem::Specification::find_all_by_name('snapshot').any?
+          return [Gem::Specification.find_by_name('snapshot').gem_dir, 'lib'].join('/')
+        else
+          return './lib'
+        end
       end
 
       def available_devices
         # The order IS important, since those names are used to check for include?
         # and the iPhone 6 is inlucded in the iPhone 6 Plus
\ No newline at end of file