From c3fb1bb5cb20f0679587a14242fb468fcf7a4d72 Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Sat, 30 Dec 2023 15:28:49 +0800
Subject: [PATCH] Fix test

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

diff --git a/internal/statements/update.go b/internal/statements/update.go
index 86287578..673d3848 100644
--- a/internal/statements/update.go
+++ b/internal/statements/update.go
@@ -555,8 +555,14 @@ func (statement *Statement) writeSetColumns(colNames []string, args []any) func(
 					return err
 				}
 			}
-			if _, err := fmt.Fprint(w, colName); err != nil {
-				return err
+			if len(statement.joins) > 0 {
+				if _, err := fmt.Fprint(w, statement.tableName, ".", colName); err != nil {
+					return err
+				}
+			} else {
+				if _, err := fmt.Fprint(w, colName); err != nil {
+					return err
+				}
 			}
 		}
 		w.Append(args...)