diff --git a/integrations/engine_test.go b/integrations/engine_test.go
index c49d3f95..997c8962 100644
--- a/integrations/engine_test.go
+++ b/integrations/engine_test.go
@@ -276,10 +276,10 @@ func TestGetColumnsComment(t *testing.T) {
 	var hasComment, noComment string
 	for _, table := range tables {
 		if table.Name == tableName {
-			col := table.GetColumn("has_comment")
+			col := table.GetColumn(testEngine.GetColumnMapper().Obj2Table("HasComment"))
 			assert.NotNil(t, col)
 			hasComment = col.Comment
-			col2 := table.GetColumn("no_comment")
+			col2 := table.GetColumn(testEngine.GetColumnMapper().Obj2Table("NoComment"))
 			assert.NotNil(t, col2)
 			noComment = col2.Comment
 			break