diff --git a/lib/jasmine/headless/coffee_template.rb b/lib/jasmine/headless/coffee_template.rb
index b6764e6..5ad8918 100644
--- a/lib/jasmine/headless/coffee_template.rb
+++ b/lib/jasmine/headless/coffee_template.rb
@@ -1,4 +1,5 @@
 require 'tilt/template'
+require 'rainbow'
 
 module Jasmine::Headless
   class CoffeeTemplate < Tilt::Template
diff --git a/spec/lib/jasmine/headless/coffee_template_spec.rb b/spec/lib/jasmine/headless/coffee_template_spec.rb
index f07e769..c4abdea 100644
--- a/spec/lib/jasmine/headless/coffee_template_spec.rb
+++ b/spec/lib/jasmine/headless/coffee_template_spec.rb
@@ -39,7 +39,7 @@ describe Jasmine::Headless::CoffeeTemplate do
       end
 
       it 'should return the cached file' do
-        subject.should include(%{<script type="text/javascript" src="#{file_path}"></script>})
+        subject.should include(%{<script from="jhw" type="text/javascript" src="#{file_path}"></script>})
       end
     end
 
@@ -47,7 +47,7 @@ describe Jasmine::Headless::CoffeeTemplate do
       let(:cache_return) { false }
 
       it 'should return the generated js' do
-        subject.should include(%{<script type="text/javascript">#{source}</script>})
+        subject.should include(%{<script from="jhw" type="text/javascript">#{source}</script>})
       end
     end
   end
diff --git a/spec/lib/jasmine/headless/template_writer_spec.rb b/spec/lib/jasmine/headless/template_writer_spec.rb
index 3b4ad0d..2a9c380 100644
--- a/spec/lib/jasmine/headless/template_writer_spec.rb
+++ b/spec/lib/jasmine/headless/template_writer_spec.rb
@@ -53,6 +53,8 @@ describe Jasmine::Headless::TemplateWriter do
 
       runner.stubs(:keep_runner).returns(true)
       runner.stubs(:runner_filename).returns(false)
+
+      Sprockets::Environment.any_instance.stubs(:find_asset).returns(stub(:body => ''))
     end
 
     let(:files_list) { Jasmine::Headless::FilesList.new }