diff --git a/auto/generate_test_runner.rb b/auto/generate_test_runner.rb
index d3bbf1f..5ddf0ab 100644
--- a/auto/generate_test_runner.rb
+++ b/auto/generate_test_runner.rb
@@ -9,7 +9,7 @@ File.expand_path(File.join(File.dirname(__FILE__),'colour_prompt'))
 class UnityTestRunnerGenerator
 
   def initialize(options = nil)
-    @options = { :includes => [] }
+    @options = { :includes => [], :framework => :unity }
     case(options)
       when NilClass then @options
       when String   then @options = UnityTestRunnerGenerator.grab_config(options)
@@ -19,7 +19,7 @@ class UnityTestRunnerGenerator
   end
   
   def self.grab_config(config_file)
-    options = { :includes => [] }
+    options = { :includes => [], :framework => :unity }
     unless (config_file.nil? or config_file.empty?)
       require 'yaml'
       yaml_guts = YAML.load_file(config_file)
@@ -27,6 +27,7 @@ class UnityTestRunnerGenerator
       options[:cexception] = 1 unless (yaml_goodness[:plugins] & ['cexception', :cexception]).empty?
       options[:coverage  ] = 1 if     (yaml_goodness[:coverage])
       options[:order]      = 1 if     (yaml_goodness[:enforce_strict_ordering])
+      options[:framework]  =          (yaml_goodness[:framework] || :unity)
       options[:includes]   <<         (yaml_goodness[:includes])
     end
     return(options)
@@ -117,7 +118,7 @@ class UnityTestRunnerGenerator
   
   def create_header(output, mocks)
     output.puts('/* AUTOGENERATED FILE. DO NOT EDIT. */')
-    output.puts('#include "unity.h"')
+    output.puts("#include \"#{@options[:framework].to_s}.h\"")
     output.puts('#include "cmock.h"') unless (mocks.empty?)
     @options[:includes].flatten.each do |includes|
       output.puts("#include \"#{includes.gsub('.h','')}.h\"")
diff --git a/auto/unity_test_summary.rb b/auto/unity_test_summary.rb
index f8035f4..8707f7a 100644
--- a/auto/unity_test_summary.rb
+++ b/auto/unity_test_summary.rb
@@ -26,7 +26,7 @@ class UnityTestSummary
   def run
     # Clean up result file names
     results = @targets.map {|target| target.gsub(/\\/,'/')}
-
+    
     # Dig through each result file, looking for details on pass/fail:   
     failure_output = []
     ignore_output = []
diff --git a/examples/rakefile_helper.rb b/examples/rakefile_helper.rb
index 6da5973..0cf386f 100644
--- a/examples/rakefile_helper.rb
+++ b/examples/rakefile_helper.rb
@@ -157,6 +157,7 @@ module RakefileHelpers
     results = Dir[results_glob]
     summary.set_targets(results)
     summary.run
+    raise "There were failures" if (summary.failures > 0)
   end
   
   def run_tests(test_files)
@@ -222,7 +223,6 @@ module RakefileHelpers
         test_results += '.testpass'
       end
       File.open(test_results, 'w') { |f| f.print output }
-      
     end
   end