From 3c15419f82fd52e4f10a53beeb56b629d3ee078b Mon Sep 17 00:00:00 2001
From: algebro <algebro@tuta.io>
Date: Thu, 19 Sep 2024 16:27:28 -0400
Subject: [PATCH] implement addAtBContract

---
 .../market/contractMarket/CamOpsContractMarket.java      | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/MekHQ/src/mekhq/campaign/market/contractMarket/CamOpsContractMarket.java b/MekHQ/src/mekhq/campaign/market/contractMarket/CamOpsContractMarket.java
index 93443e99af8..7a543deb961 100644
--- a/MekHQ/src/mekhq/campaign/market/contractMarket/CamOpsContractMarket.java
+++ b/MekHQ/src/mekhq/campaign/market/contractMarket/CamOpsContractMarket.java
@@ -36,6 +36,12 @@ public CamOpsContractMarket() {
 
     @Override
     public AtBContract addAtBContract(Campaign campaign) {
+        Optional<AtBContract> c = generateContract(campaign);
+        if (c.isPresent()) {
+            AtBContract atbContract = c.get();
+            contracts.add(atbContract);
+            return atbContract;
+        }
         return null;
     }
 
@@ -58,8 +64,7 @@ public void generateContractOffers(Campaign campaign, boolean newCampaign) {
             rollNegotiation(negotiationSkill, ratingMod + contractMods.offersMod) - BASE_NEGOTIATION_TARGET);
 
         for (int i = 0; i < numOffers; i++) {
-            Optional<AtBContract> c = generateContract(campaign);
-            c.ifPresent(contract -> contracts.add(contract));
+            addAtBContract(campaign);
         }
         updateReport(campaign);
     }