diff --git a/src/admin/problem/components/SQLTestcaseEditor.vue b/src/admin/problem/components/SQLTestcaseEditor.vue
index f5dddbb..30665e8 100644
--- a/src/admin/problem/components/SQLTestcaseEditor.vue
+++ b/src/admin/problem/components/SQLTestcaseEditor.vue
@@ -288,9 +288,7 @@ INSERT INTO ...;"
           <p class="previewTitle">期望结果</p>
           <SQLDataTable
             v-if="expectedQuery(s.display)"
-            :columns="
-              expectedQuery(s.display)!.columns.map((name) => ({ name }))
-            "
+            :columns="expectedQuery(s.display)!.columns"
             :rows="expectedQuery(s.display)!.rows"
             :total-rows="expectedQuery(s.display)!.total_rows"
             :truncated="expectedQuery(s.display)!.truncated"
diff --git a/src/oj/problem/components/ProblemContent.vue b/src/oj/problem/components/ProblemContent.vue
index 7b4524a..afcafe7 100644
--- a/src/oj/problem/components/ProblemContent.vue
+++ b/src/oj/problem/components/ProblemContent.vue
@@ -333,7 +333,7 @@ function type(status: ProblemStatus) {
       </p>
       <template v-if="sqlExpectedQuery">
         <SQLDataTable
-          :columns="sqlExpectedQuery.columns.map((name) => ({ name }))"
+          :columns="sqlExpectedQuery.columns"
           :rows="sqlExpectedQuery.rows"
           :total-rows="sqlExpectedQuery.total_rows"
           :truncated="sqlExpectedQuery.truncated"
diff --git a/src/utils/types.ts b/src/utils/types.ts
index 525b283..db11c1f 100644
--- a/src/utils/types.ts
+++ b/src/utils/types.ts
@@ -77,7 +77,7 @@ export interface SQLDisplay {
   tables: SQLDisplayTable[]
   expected:
     | {
-        columns: string[]
+        columns: SQLDisplayColumn[]
         rows: (string | number | null)[][]
         total_rows: number
         truncated: boolean