diff --git a/backup/restore_to_primary.sh b/backup/restore_to_primary.sh
index edd5bdd..4b04403 100755
--- a/backup/restore_to_primary.sh
+++ b/backup/restore_to_primary.sh
@@ -24,7 +24,7 @@ SCRIPT_DIR="$(cd "$(dirname "$0")/.." && pwd)"
 source "${SCRIPT_DIR}/lib/common.sh"
 
 load_env
-require_vars UNRAID_IP UNRAID_SSH_USER UNRAID_GITEA_DATA_PATH \
+require_vars UNRAID_IP UNRAID_SSH_USER UNRAID_GITEA_DATA_PATH UNRAID_COMPOSE_DIR \
              GITEA_INTERNAL_URL GITEA_ADMIN_USER GITEA_ADMIN_PASSWORD \
              GITEA_DB_TYPE
 if [[ "${GITEA_DB_TYPE}" != "sqlite3" ]]; then
@@ -32,6 +32,7 @@ if [[ "${GITEA_DB_TYPE}" != "sqlite3" ]]; then
 fi
 
 DATA_PATH="$UNRAID_GITEA_DATA_PATH"
+COMPOSE_DIR="${UNRAID_COMPOSE_DIR}/gitea"
 
 # ---------------------------------------------------------------------------
 # Parse arguments
@@ -94,7 +95,7 @@ log_success "Archive ready on Unraid: ${UNRAID_ARCHIVE}"
 # Step 2: Stop Gitea container
 # ---------------------------------------------------------------------------
 log_step 2 "Stopping Gitea container..."
-ssh_exec UNRAID "cd '${DATA_PATH}' && docker compose down 2>/dev/null || docker-compose down" || true
+ssh_exec UNRAID "cd '${COMPOSE_DIR}' && docker compose down 2>/dev/null || docker-compose down" || true
 log_success "Gitea container stopped"
 
 # ---------------------------------------------------------------------------
@@ -172,7 +173,7 @@ ssh_exec UNRAID "rm -f '${UNRAID_ARCHIVE}'"
 # Step 5: Restart Gitea container
 # ---------------------------------------------------------------------------
 log_step 5 "Starting Gitea container..."
-ssh_exec UNRAID "cd '${DATA_PATH}' && docker compose up -d 2>/dev/null || docker-compose up -d"
+ssh_exec UNRAID "cd '${COMPOSE_DIR}' && docker compose up -d 2>/dev/null || docker-compose up -d"
 log_success "Gitea container started"
 
 # ---------------------------------------------------------------------------