2 files changed, 12 insertions, 0 deletions

diff --git a/src/AssetManager.cpp b/src/AssetManager.cpp
index 97833f5..5fa7990 100644
--- a/src/AssetManager.cpp
+++ b/src/AssetManager.cpp
@@ -455,6 +455,10 @@ void ParseBlockModels() {
 					}
 					textureName.insert(0, "minecraft/textures/");
 					texture = AssetManager::GetTexture(textureName);
+					if (textureName.find("water_flow") != std::string::npos)
+						texture.h /= 32.0f;
+					if (textureName.find("lava_flow") != std::string::npos)
+						texture.h /= 16.0f;
 
 					if (!(face.second.uv == BlockModel::ElementData::FaceData::Uv{ 0,16,0,16 }) && !(face.second.uv == BlockModel::ElementData::FaceData::Uv{ 0,0,0,0 })
 						&& !(face.second.uv == BlockModel::ElementData::FaceData::Uv{ 0,0,16,16 })) {
diff --git a/src/Block.cpp b/src/Block.cpp
index 6756960..3de1901 100644
--- a/src/Block.cpp
+++ b/src/Block.cpp
@@ -44,6 +44,14 @@ std::pair<std::string, std::string> TransformBlockIdToBlockStateName(BlockId blo
 		};
 		return ids[blockId.state];
 	}
+	case 8:
+	case 9: {
+		return std::make_pair("water", "normal");
+	}
+	case 10:
+	case 11: {
+		return std::make_pair("lava", "normal");
+	}
 	case 12: {
 		if (blockId.state > 1)
 			break;