diff --git a/configs/sqlite3.go b/configs/sqlite3.go
index 35202f1..8569ed8 100644
--- a/configs/sqlite3.go
+++ b/configs/sqlite3.go
@@ -9,10 +9,6 @@ import (
 )
 
 func init() {
-	// 設置日誌顯示文件名和行號
-	log.SetFlags(log.Lshortfile | log.LstdFlags)
-
-	// 原生golang 創建 data 目錄不存在則創建
 	if _, err := os.Stat("data"); os.IsNotExist(err) {
 		os.Mkdir("data", os.ModePerm)
 	}
diff --git a/main.go b/main.go
index 6379d41..8904760 100644
--- a/main.go
+++ b/main.go
@@ -16,6 +16,7 @@ import (
 
 func main() {
 	runtime.GOMAXPROCS(runtime.NumCPU())
+	log.SetFlags(log.Lshortfile | log.LstdFlags)
 
 	r := mux.NewRouter()
 
diff --git a/routers/models.go b/routers/models.go
index 21608f3..38274d7 100644
--- a/routers/models.go
+++ b/routers/models.go
@@ -105,8 +105,9 @@ func ModelItemGet(w http.ResponseWriter, r *http.Request) {
 
 func ModelItemPatch(w http.ResponseWriter, r *http.Request) {
 	var model = models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
-	if err := configs.ORMDB().Take(&model, utils.ParamInt(mux.Vars(r)["id"], 0)); err != nil {
+	if err := configs.ORMDB().Take(&model, utils.ParamInt(mux.Vars(r)["id"], 0)).Error; err != nil {
 		w.WriteHeader(http.StatusNotFound)
+		w.Write([]byte(err.Error()))
 		return
 	}
 
@@ -133,9 +134,9 @@ func ModelItemPatch(w http.ResponseWriter, r *http.Request) {
 	if model_new.Status != "" && model_new.Status != model.Status {
 		model.Status = model_new.Status
 		// 如果狀態被改變爲 ready, 將模型發送到訓練隊列
-		if model.Status == "ready" {
-			//model.SendToTrain()
-		}
+		//if model.Status == "ready" {
+		//	model.SendToTrain()
+		//}
 	}
 	if model_new.Image != "" && model_new.Image != model.Image {
 		model.Image = model_new.Image
diff --git a/routers/sessions.go b/routers/sessions.go
index c839db4..21144c2 100644
--- a/routers/sessions.go
+++ b/routers/sessions.go
@@ -107,7 +107,7 @@ func SessionsItemGet(w http.ResponseWriter, r *http.Request) {
 // 更新會話
 func SessionsItemPatch(w http.ResponseWriter, r *http.Request) {
 	session := models.Session{ID: mux.Vars(r)["session_id"]}
-	if err := configs.ORMDB().Model(&session).Updates(GetForm(r)); err != nil {
+	if err := configs.ORMDB().Model(&session).Updates(GetForm(r)).Error; err != nil {
 		w.WriteHeader(http.StatusNotFound)
 		w.Write([]byte("404 - Not Found"))
 		return