1 files changed, 11 insertions, 3 deletions

diff --git a/api/internal/controllers/user.go b/api/internal/controllers/user.go
index dbb09cf..fa9617a 100644
--- a/api/internal/controllers/user.go
+++ b/api/internal/controllers/user.go
@@ -15,7 +15,11 @@ func GetUser(c *gin.Context) {
         c.JSON(http.StatusOK, gin.H{"message": "User found"})
 }
 func GetUserPreferences(c *gin.Context) {
-        db := database.EstablishDBConnection()
+        db, err := database.EstablishDBConnection()
+        if err != nil {
+                c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                return
+        }
         defer func(db *sql.DB) {
                 err := db.Close()
                 if err != nil {
@@ -41,7 +45,11 @@ func GetUserPreferences(c *gin.Context) {
 }
 
 func UpdateUserPreferences(c *gin.Context) {
-        db := database.EstablishDBConnection()
+        db, err := database.EstablishDBConnection()
+        if err != nil {
+                c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+                return
+        }
         defer func(db *sql.DB) {
                 err := db.Close()
                 if err != nil {
@@ -58,7 +66,7 @@ func UpdateUserPreferences(c *gin.Context) {
 
         log.Printf("newPreferences: %v", newPreferences)
 
-        _, err := db.Exec("UPDATE Preferences SET color = ?, size_id = ? WHERE id = ?", newPreferences.Color, newPreferences.SizeID, newPreferences.ID)
+        _, err = db.Exec("UPDATE Preferences SET color = ?, size_id = ? WHERE id = ?", newPreferences.Color, newPreferences.SizeID, newPreferences.ID)
         if err != nil {
                 c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
                 return