test/test_kindlegen.rb in kindlegen-3.1.0 vs test/test_kindlegen.rb in kindlegen-3.1.1

@@ -7,10 +7,10 @@
 class KindlegenTest < Test::Unit::TestCase
   def test_gem_install
     kindlegen_lib_dir = nil
     gem_version = File.read(File.join(KINDLEGEN_PROJECT_DIR, 'lib/kindlegen/version.rb')).match(/VERSION = ["'](.*?)["']/)[1]
     gem_file = File.join(KINDLEGEN_PROJECT_DIR, 'pkg', %(kindlegen-#{gem_version}.gem))
-	 result = Gem::Installer.at(gem_file).install rescue Gem::Installer.new(gem_file).install
+	 result = Gem::Installer.at(gem_file).install rescue Gem::Installer.new(gem_file).install rescue Gem::Installer.new(Gem::Package.new gem_file).install
     begin
       require 'kindlegen'
     rescue ::LoadError
       Dir.glob(result.lib_dirs_glob).each do |path|
         $:.unshift path