diff --git a/ProjectSourceCode/src/index.js b/ProjectSourceCode/src/index.js
index 3f4b137..f9dc19f 100644
--- a/ProjectSourceCode/src/index.js
+++ b/ProjectSourceCode/src/index.js
@@ -295,12 +295,12 @@ app.post('/favteam/add', async (req, res, next) => {
 });
 app.post('/favteam/remove', async (req, res) => {
   try {
-    const { userID, teamName } = req.body;
+    const { userID, teamID } = req.body;
 
     // Check if the team exists for the user
     const existingTeam = await db.oneOrNone(
-      'SELECT * FROM FavoriteTeams WHERE UserID = $1 AND TeamName = $2',
-      [userID, teamName]
+      'SELECT * FROM FavoriteTeams WHERE UserID = $1 AND TeamID = $2',
+      [userID, teamID]
     );
 
     // If the team does not exist for the user, return a 404 error
@@ -310,8 +310,8 @@ app.post('/favteam/remove', async (req, res) => {
 
     // Remove the favorite team from the database
     await db.none(
-      'DELETE FROM FavoriteTeams WHERE UserID = $1 AND TeamName = $2',
-      [userID, teamName]
+      'DELETE FROM FavoriteTeams WHERE UserID = $1 AND TeamID = $2',
+      [userID, teamID]
     );
 
     console.log('Favorite team removed successfully.');
diff --git a/ProjectSourceCode/src/resources/js/club-page/favorite-button.js b/ProjectSourceCode/src/resources/js/club-page/favorite-button.js
index 49fa031..5697a60 100644
--- a/ProjectSourceCode/src/resources/js/club-page/favorite-button.js
+++ b/ProjectSourceCode/src/resources/js/club-page/favorite-button.js
@@ -9,23 +9,9 @@ document.addEventListener("DOMContentLoaded", function() {
             
             if (favoriteButton.src.includes("/favorited.png")) {
                 removeFavoriteTeam(userID, teamID)
-                .then((rmv_status) => {
-                    console.log(rmv_status);
-                    if (rmv_status === 200) {
-                        favoriteButton.src = "/img/club-page/unfavorited.png";
-                    }
-                })
-                .catch(error => {
-                        console.error('Error removing favorite team:', error);
-                });
-            } else {
+            }
+            else {
                 addFavoriteTeam(userID, teamID, teamName, teamLogo)
-                    .then(() => {
-                        favoriteButton.src = "/img/club-page/favorited.png";
-                    })
-                    .catch(error => {
-                        console.error('Error adding favorite team:', error);
-                    });
             }
         });
     }
@@ -49,8 +35,11 @@ async function addFavoriteTeam(userID, teamID, teamName, teamLogo){
         if (!response.ok) {
             throw new Error('Failed to add favorite team');
         }
-
+        //Changes button if favorite team is added//
         console.log('New favorite team added successfully.');
+        var favoriteButton = document.getElementById("club-favorite-button");
+        favoriteButton.src = "/img/club-page/favorited.png";
+
     } catch (error) {
         console.error('Error adding favorite team:', error);
 
@@ -69,6 +58,10 @@ async function removeFavoriteTeam(userID, teamID) {
             })
         });
         console.log('Favorite team removed successfully.');
+        //Change button source//
+        var favoriteButton = document.getElementById("club-favorite-button");
+        favoriteButton.src = "/img/club-page/unfavorited.png";
+
     } catch (error) {
         console.error('Error removing favorite team:', error);
     }