diff --git a/dialect/sql/schema/schema.go b/dialect/sql/schema/schema.go
index a1df7111c..ef590853d 100644
--- a/dialect/sql/schema/schema.go
+++ b/dialect/sql/schema/schema.go
@@ -111,8 +111,7 @@ func (t *Table) index(name string) (*Index, bool) {
 	c, ok := t.column(name)
 	if !ok {
 		// Postgres naming convention for unique constraint.
-		name = strings.TrimSuffix(name, "_key")
-		c, ok = t.column(name)
+		c, ok = t.column(strings.TrimSuffix(name, "_key"))
 	}
 	if ok && c.Unique {
 		return &Index{Name: name, Unique: c.Unique, Columns: []*Column{c}, columns: []string{c.Name}}, true