Fix hydration bug

Fixes issue #2