Fixed water system not updating properly in edit mode