From 30e9cb9c023dcfcbda13de9c775261e87d9f0ba1 Mon Sep 17 00:00:00 2001
From: lowercase-donkey <lowercasedonkey@gmail.com>
Date: Mon, 6 May 2019 19:24:39 -0400
Subject: [PATCH] Fix madam income not recorded

---
 src/endWeek/saWhore.js | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/endWeek/saWhore.js b/src/endWeek/saWhore.js
index 12ce38960d7..efde335e33d 100644
--- a/src/endWeek/saWhore.js
+++ b/src/endWeek/saWhore.js
@@ -41,7 +41,7 @@ window.saWhore = (function saWhore() {
 
 		gatherStatistics(slave);
 		updateNonSlaveVariables(slave); // must be run before applyFSDecoration() or you will face NaNs
-		if (slave.assignment === "work in the brothel") {
+		if (slave.assignment === "work in the brothel" || slave.assignment === "be the Madam") {
 			// By being at the end, every slave after the first will get a bonus. By moving it up, the first can enjoy it too. slaveJobValues() checks Edo Rivalist, so here we are.
 			applyFSDecoration(slave);
 		}
@@ -76,7 +76,12 @@ window.saWhore = (function saWhore() {
 	 * */
 	function gatherStatistics(slave) {
 		/* Statistics gathering */
-		const facility = slave.assignment === Job.BROTHEL ? V.facility.brothel : undefined;
+		let facility;
+		if (slave.assignment === Job.BROTHEL || slave.assignment === Job.MADAM) {
+			facility = V.facility.brothel;
+		} else {
+			facility = undefined;
+		}
 		T.incomeStats = getSlaveStatisticData(slave, facility);
 	}
 
-- 
GitLab