Formation water treatment unit