Fix hydration mode setup