diff --git a/Winget-AutoUpdate/functions/Update-App.ps1 b/Winget-AutoUpdate/functions/Update-App.ps1
index 4d93c8b..a72951f 100644
--- a/Winget-AutoUpdate/functions/Update-App.ps1
+++ b/Winget-AutoUpdate/functions/Update-App.ps1
@@ -19,6 +19,9 @@ Function Update-App ($app) {
     #Run Winget Upgrade command
     & $Winget upgrade --id $($app.Id) --all --accept-package-agreements --accept-source-agreements -h | Tee-Object -file $LogFile -Append
 
+    #Set mods to apply as an upgrade
+    $ModsMode = "Upgrade"
+
     #Check if application updated properly
     $CheckOutdated = Get-WingetOutdatedApps
     $FailedToUpgrade = $false
@@ -28,6 +31,9 @@ Function Update-App ($app) {
             #If app failed to upgrade, run Install command
             & $Winget install --id $($app.Id) --accept-package-agreements --accept-source-agreements -h | Tee-Object -file $LogFile -Append
 
+            #Set mods to apply as an install
+            $ModsMode = "Install"
+
             #Check if application installed properly
             $CheckOutdated2 = Get-WingetOutdatedApps
             foreach ($CheckApp2 in $CheckOutdated2) {
@@ -42,10 +48,14 @@ Function Update-App ($app) {
 
         #Check if mods exist for install/upgrade
         $ModsInstall, $ModsUpgrade = Test-Mods $($app.Id)
-        if ($ModsUpgrade) {
+        if ($ModsMode = "Upgrade") {
             Write-Log "Modifications for $($app.Id) after upgrade are being applied..." "Yellow"
             & "$ModsUpgrade"
         }
+        elseif ($ModsMode = "Install") {
+            Write-Log "Modifications for $($app.Id) after install are being applied..." "Yellow"
+            & "$ModsInstall"
+        }
     }
 
     Write-Log "##########   WINGET UPGRADE PROCESS FINISHED FOR APPLICATION ID '$($App.Id)'   ##########" "Gray"