diff --git a/lib/html2js.js b/lib/html2js.js
index cc734b0..3ac5e1c 100644
--- a/lib/html2js.js
+++ b/lib/html2js.js
@@ -1,9 +1,9 @@
var util = require('util');
-var TEMPLATE = 'angular.module(\'%s\', []).run(function($templateCache) {\n' +
+var TEMPLATE = 'angular.module(\'%s\', []).run([\'$templateCache\', function($templateCache) {\n' +
' $templateCache.put(\'%s\',\n \'%s\');\n' +
- '});\n';
+ '}]);\n';
var SINGLE_MODULE_TPL = '(function(module) {\n' +
'try {\n' +
@@ -11,9 +11,9 @@ var SINGLE_MODULE_TPL = '(function(module) {\n' +
'} catch (e) {\n' +
' module = angular.module(\'%s\', []);\n' +
'}\n' +
- 'module.run(function($templateCache) {\n' +
+ 'module.run([\'$templateCache\', function($templateCache) {\n' +
' $templateCache.put(\'%s\',\n \'%s\');\n' +
- '});\n' +
+ '}]);\n' +
'})();\n';
var escapeContent = function(content) {
diff --git a/test/helpers/template_cache.coffee b/test/helpers/template_cache.coffee
index bcb3c87..0c89d0c 100644
--- a/test/helpers/template_cache.coffee
+++ b/test/helpers/template_cache.coffee
@@ -7,7 +7,8 @@ module.exports = (chai, utils) ->
templates = @templates = {}
run: (block) ->
- block
+ # Block is an annotation array ["$templateCache", function($templateCache) {..}]
+ block[1]
put: (id, content) =>
@templates[id] = content