diff --git a/src/app.js b/src/app.js
index dc23573bb5..79a7143db5 100644
--- a/src/app.js
+++ b/src/app.js
@@ -157,7 +157,7 @@ var run = function () {
 
   // ----------------- editor ----------------------
 
-  var editor = new Editor('input')
+  var editor = new Editor(document.getElementById('input'))
 
   // ----------------- tabbed menu -------------------
   $('#options li').click(function (ev) {
diff --git a/src/app/editor.js b/src/app/editor.js
index 9f48074b27..e5f4e59e0a 100644
--- a/src/app/editor.js
+++ b/src/app/editor.js
@@ -8,7 +8,7 @@ require('../mode-solidity.js')
 
 function Editor (editorElement) {
   var editor = ace.edit(editorElement)
-  document.getElementById(editorElement).editor = editor // required to access the editor during tests
+  editorElement.editor = editor // required to access the editor during tests
   var event = new EventManager()
   this.event = event
   var sessions = {}