diff --git a/cooling_lava.lua b/cooling_lava.lua
index 035c4fd..c57da0a 100644
--- a/cooling_lava.lua
+++ b/cooling_lava.lua
@@ -221,4 +221,21 @@ minetest.register_abm({
 	action = function(...)
 		cool_lava_source(...)
 	end,
-})
\ No newline at end of file
+})
+
+-- new water types to work with lava cooling
+dynamic_liquid.add_to_water_nodes = function(new_source, new_flowing)
+	-- Flowing liquid
+	table.insert(dynamic_lava_flowing_destroys, new_flowing)
+	table.insert(dynamic_lava_source_destroys, new_flowing)
+
+	table.insert(all_flowing_nodes, new_flowing)
+	table.insert(all_source_nodes, new_flowing)
+
+	--Source liquid
+	table.insert(dynamic_lava_source_destroys, new_source)
+	table.insert(dynamic_cools_lava_flowing, new_source)
+	table.insert(dynamic_cools_lava_source, new_source)
+
+	table.insert(all_source_nodes, new_source)
+end