diff --git a/www/code/main.js b/www/code/main.js
index 9cf46b8d6..554215681 100644
--- a/www/code/main.js
+++ b/www/code/main.js
@@ -297,7 +297,7 @@ define([
                         var mime = CodeMirror.findModeByMIME(file.type);
 
                         if (!mime) {
-                            var ext = /.+\.([^.]+)$/.exec(val);
+                            var ext = /.+\.([^.]+)$/.exec(file.name);
                             if (ext[1]) {
                                 mode = CodeMirror.findModeByExtension(ext[1]);
                             }