diff --git a/extension/src/experiments/index.ts b/extension/src/experiments/index.ts
index 4133f1fe17..1bec430040 100644
--- a/extension/src/experiments/index.ts
+++ b/extension/src/experiments/index.ts
@@ -191,7 +191,9 @@ export class Experiments extends BaseRepository<TableData> {
 
   public async addSort() {
     const columns = this.columns.getTerminalNodes()
-    const sortToAdd = await pickSortToAdd(columns)
+    const columnLikes = addStarredToColumns(columns)
+
+    const sortToAdd = await pickSortToAdd(columnLikes)
     if (!sortToAdd) {
       return
     }
diff --git a/extension/src/experiments/model/sortBy/quickPick.ts b/extension/src/experiments/model/sortBy/quickPick.ts
index 4d4c74f1b4..9ccde4f998 100644
--- a/extension/src/experiments/model/sortBy/quickPick.ts
+++ b/extension/src/experiments/model/sortBy/quickPick.ts
@@ -3,10 +3,10 @@ import { definedAndNonEmpty } from '../../../util/array'
 import { quickPickManyValues, quickPickValue } from '../../../vscode/quickPick'
 import { Title } from '../../../vscode/title'
 import { Toast } from '../../../vscode/toast'
+import { ColumnLike } from '../../columns/like'
 import { pickFromColumnLikes } from '../../columns/quickPick'
-import { Column } from '../../webview/contract'
 
-export const pickSortToAdd = async (columns: Column[]) => {
+export const pickSortToAdd = async (columns: ColumnLike[] | undefined) => {
   const picked = await pickFromColumnLikes(columns, {
     title: Title.SELECT_PARAM_OR_METRIC_SORT
   })