         notebook: notebook
         input_cell: input_cell
         set_code: (code) ->
-          cell = add_input_cell notebook, code: code, after: run_context.cell
+          cell = add_input_cell notebook, code: code, after: run_context.input_cell
           focus_cell cell
         run: (code) ->
-          cell = add_input_cell notebook, code: code, after: run_context.cell
+          cell = add_input_cell notebook, code: code, after: run_context.input_cell
           run cell
         previously_run: -> input_cell_at_offset(input_cell, -1).editor.getValue()
         export_notebook: -> export_notebook input_cell

          unless version?
            run_context.fns.error "File #{file.filename} isn't a lead.js notebook"
            return
          import_notebook run_context.notebook, run_context.cell, imported, options

    load_file = (run_context, file) ->
      if file.type.indexOf('image') < 0

      if file.type.indexOf('image') < 0
        [prefix..., extension] = file.filename.split '.'
        if extension is 'coffee'
          cell = add_input_cell run_context.notebook, code: file.content, after: run_context.input_cell
          if options.run
            run cell
        else
