From 55d79602e326e104fe8ddb00bbba656c673201ac Mon Sep 17 00:00:00 2001
From: Blue Staggo <owengrenney@gmail.com>
Date: Mon, 26 May 2025 18:10:09 +0100
Subject: [PATCH] Fix river generation

---
 .../world/biome/provider/fractal/layers/MixRiverLayer.java       | 1 +
 .../modernerbeta/fabric/data/ModernBetaTagProviderBiome.java     | 1 +
 2 files changed, 2 insertions(+)

diff --git a/common/src/main/java/mod/bluestaggo/modernerbeta/world/biome/provider/fractal/layers/MixRiverLayer.java b/common/src/main/java/mod/bluestaggo/modernerbeta/world/biome/provider/fractal/layers/MixRiverLayer.java
index 619a5447..d1c4a55f 100644
--- a/common/src/main/java/mod/bluestaggo/modernerbeta/world/biome/provider/fractal/layers/MixRiverLayer.java
+++ b/common/src/main/java/mod/bluestaggo/modernerbeta/world/biome/provider/fractal/layers/MixRiverLayer.java
@@ -76,6 +76,7 @@ public class MixRiverLayer extends SingleParentLayer {
 
     @Override
     protected void addPossibleBiomes(Set<ExtendedBiomeId> biomes) {
+        biomes.add(ExtendedBiomeId.RIVER);
         for (Map.Entry<ExtendedBiomeId, ExtendedBiomeId> entry : this.biomeSpecificRivers.entrySet()) {
             if (biomes.contains(entry.getKey())) {
                 biomes.add(entry.getValue());
diff --git a/fabric/src/main/java/mod/bluestaggo/modernerbeta/fabric/data/ModernBetaTagProviderBiome.java b/fabric/src/main/java/mod/bluestaggo/modernerbeta/fabric/data/ModernBetaTagProviderBiome.java
index 382e1737..05f3f228 100644
--- a/fabric/src/main/java/mod/bluestaggo/modernerbeta/fabric/data/ModernBetaTagProviderBiome.java
+++ b/fabric/src/main/java/mod/bluestaggo/modernerbeta/fabric/data/ModernBetaTagProviderBiome.java
@@ -398,6 +398,7 @@ public class ModernBetaTagProviderBiome extends FabricTagProvider<Biome> {
             );
 
         getOrCreateTagBuilder(HEIGHT_CONFIG_RIVER)
+            .forceAddTag(BiomeTags.IS_RIVER)
             .add(
                 BiomeKeys.RIVER,
                 BiomeKeys.FROZEN_RIVER
-- 
knotmirror 0.0.1