diff --git a/lib/typescript-node.rb b/lib/typescript-node.rb
index 7d561ba..c8039e3 100644
--- a/lib/typescript-node.rb
+++ b/lib/typescript-node.rb
@@ -39,8 +39,8 @@ def compile_file(source_file, *tsc_options)
       # Compile TypeScript to JavaScript.
       # @param [String] source TypeScript to be compiled
       # @return [String] Resulted JavaScript
-      def compile(source, *tsc_options)
-        js_file = Tempfile.new(%w(typescript-node .ts))
+      def compile(source, extension, *tsc_options)
+        js_file = Tempfile.new(["typescript-node", extension])
         begin
           js_file.write(source)
           js_file.close