From df2e20b4cef49ef17aedd91879b6413a83feb2ac Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Sat, 30 Dec 2023 16:43:55 +0800
Subject: [PATCH] Fix test

---
 internal/statements/update.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/internal/statements/update.go b/internal/statements/update.go
index cc586de2..91e09bfd 100644
--- a/internal/statements/update.go
+++ b/internal/statements/update.go
@@ -583,8 +583,12 @@ func (statement *Statement) writeSetColumns(colNames []string, args []any) func(
 					return err
 				}
 			}
-			if statement.dialect.URI().DBType == schemas.MSSQL && len(statement.joins) > 0 {
-				if _, err := fmt.Fprint(w, statement.tableName, ".", colName); err != nil {
+			if statement.dialect.URI().DBType != schemas.SQLITE && len(statement.joins) > 0 {
+				tbName := statement.TableAlias
+				if tbName == "" {
+					tbName = statement.TableName()
+				}
+				if _, err := fmt.Fprint(w, tbName, ".", colName); err != nil {
 					return err
 				}
 			} else {