diff --git a/extensions/ql-vscode/src/view/model-editor/MethodRow.tsx b/extensions/ql-vscode/src/view/model-editor/MethodRow.tsx
index b2b6a5e8b30..296204d2fbd 100644
--- a/extensions/ql-vscode/src/view/model-editor/MethodRow.tsx
+++ b/extensions/ql-vscode/src/view/model-editor/MethodRow.tsx
@@ -269,7 +269,7 @@ const ModelableMethodRow = forwardRef<HTMLElement | undefined, MethodRowProps>(
                 </DataGridCell>
                 {viewState.showMultipleModels && (
                   <DataGridCell>
-                    {index === modeledMethods.length - 1 ? (
+                    {index === 0 ? (
                       <CodiconRow
                         appearance="icon"
                         aria-label="Add new model"
diff --git a/extensions/ql-vscode/src/view/model-editor/__tests__/MethodRow.spec.tsx b/extensions/ql-vscode/src/view/model-editor/__tests__/MethodRow.spec.tsx
index 2d15d213647..c6821675e3c 100644
--- a/extensions/ql-vscode/src/view/model-editor/__tests__/MethodRow.spec.tsx
+++ b/extensions/ql-vscode/src/view/model-editor/__tests__/MethodRow.spec.tsx
@@ -350,7 +350,7 @@ describe(MethodRow.name, () => {
     expect(removeButton?.getElementsByTagName("input")[0]).toBeEnabled();
   });
 
-  it("shows add model button on last row and remove model button on all other rows", async () => {
+  it("shows add model button on first row and remove model button on all other rows", async () => {
     render({
       modeledMethods: [
         { ...modeledMethod, type: "source" },
@@ -401,7 +401,7 @@ describe(MethodRow.name, () => {
     ]);
   });
 
-  it("can delete the first modeled method", async () => {
+  it("cannot delete the first modeled method (but delete second instead)", async () => {
     render({
       modeledMethods: [
         { ...modeledMethod, type: "source" },
@@ -420,7 +420,7 @@ describe(MethodRow.name, () => {
 
     expect(onChange).toHaveBeenCalledTimes(1);
     expect(onChange).toHaveBeenCalledWith(method.signature, [
-      { ...modeledMethod, type: "sink" },
+      { ...modeledMethod, type: "source" },
       { ...modeledMethod, type: "none" },
       { ...modeledMethod, type: "summary" },
     ]);
@@ -441,7 +441,7 @@ describe(MethodRow.name, () => {
     });
 
     onChange.mockReset();
-    await userEvent.click(screen.getAllByLabelText("Remove model")[2]);
+    await userEvent.click(screen.getAllByLabelText("Remove model")[1]);
 
     expect(onChange).toHaveBeenCalledTimes(1);
     expect(onChange).toHaveBeenCalledWith(method.signature, [