Cleanup

src/console/widget.ts

@@ -228,6 +228,7 @@ class ConsoleWidget extends Widget {
             })[0];
             if (setNextInput) {
               let text = (setNextInput as any).text;
+              // Ignore the `replace` value and always set the next prompt.
               this.prompt.model.source = text;
             }
           }

src/notebook/notebook/actions.ts

@@ -904,13 +904,8 @@ namespace Private {
         let cell = parent.model.factory.createCodeCell();
         cell.source = text;
         let cells = parent.model.cells;
-        for (let i = 0; i < cells.length; i++) {
-          let widget = parent.childAt(i);
-          if (widget === child) {
-            cells.insert(i + 1, cell);
-            break;
-          }
-        }
+        let i = cells.indexOf(child.model);
+        cells.insert(i + 1, cell);
       }
     }
   }