lib/generators/cucumber/templates/Rakefile in singem-0.0.6 vs lib/generators/cucumber/templates/Rakefile in singem-0.0.7

@@ -41,13 +41,13 @@
   s.description = s.summary
   s.author = AUTHOR
   s.email = EMAIL
   s.homepage = HOMEPAGE
 
-  manifest = Bundler::Environment.load(File.dirname(__FILE__) + '/Gemfile')
-  manifest.dependencies.each do |d|
-    next if d.only && d.only.include?('test')
-    s.add_dependency(d.name, d.version)
+  bundle = Bundler::Bundle.load(File.dirname(__FILE__) + '/Gemfile')
+  bundle.environment.dependencies.each do |dep|
+    next unless dep.only && dep.only.include?('runtime')
+    s.add_dependency(dep.name, dep.version.to_s)
   end
 
   s.require_path = 'lib'
   s.files = %w(LICENSE README.md Rakefile TODO) + Dir.glob("{lib,features}/**/*")
 end