The simplest solution would be to charge a fair price for water since the vast majority of water is used for farming. But since most corporate farms have multi-decade long water contracts, this isn't possible (aka you must legislate, which would also benefit corporate farms).