diff --git a/src/utility/assignWidgets.tw b/src/utility/assignWidgets.tw
index 7744aff956f7b4c5fbcc56fc9944ab12a9f36e9a..0edfafbac9bf6aa729a731c951757fd59390dcc3 100644
--- a/src/utility/assignWidgets.tw
+++ b/src/utility/assignWidgets.tw
@@ -130,7 +130,7 @@
 			<<set $BrothiIDs.delete(_wID), $brothelSlaves-->>
 		<<case "be confined in the cellblock" "cellblock">>
 			<<set $args[0].assignment = "rest">>
-			<<if ($slaves[$i].inflation > 0)>>
+			<<if ($slaves[_wi].inflation > 0)>>
 				<<set $args[0].inflation = 0, $args[0].inflationType = "none", $args[0].inflationMethod = 0>>
 				<<SetBellySize $slaves[$i]>>
 			<</if>>