From 3e9e6aab2bbc2bdf2c0e3705930f199bd45cb932 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tomasz=20Zieli=C5=84ski?=
 <tomaszzielinskijedenzwielu@gmail.com>
Date: Wed, 7 Jun 2023 15:08:50 +0200
Subject: [PATCH] Fix prison distribution

---
 lib/rmg/modificators/ObjectDistributor.cpp | 1 +
 lib/rmg/modificators/TreasurePlacer.cpp    | 1 +
 2 files changed, 2 insertions(+)

diff --git a/lib/rmg/modificators/ObjectDistributor.cpp b/lib/rmg/modificators/ObjectDistributor.cpp
index 0c3b0a356..92570b791 100644
--- a/lib/rmg/modificators/ObjectDistributor.cpp
+++ b/lib/rmg/modificators/ObjectDistributor.cpp
@@ -29,6 +29,7 @@ VCMI_LIB_NAMESPACE_BEGIN
 void ObjectDistributor::process()
 {
 	distributeLimitedObjects();
+	distributePrisons();
 	distributeSeerHuts();
 }
 
diff --git a/lib/rmg/modificators/TreasurePlacer.cpp b/lib/rmg/modificators/TreasurePlacer.cpp
index 88d9f35e2..9d2e7de17 100644
--- a/lib/rmg/modificators/TreasurePlacer.cpp
+++ b/lib/rmg/modificators/TreasurePlacer.cpp
@@ -41,6 +41,7 @@ void TreasurePlacer::process()
 
 void TreasurePlacer::init()
 {
+	maxPrisons = 0; //Should be in the constructor, but we use macro for that
 	DEPENDENCY(ObjectManager);
 	DEPENDENCY(ConnectionsPlacer);
 	POSTFUNCTION(RoadPlacer);