Bug fixes

Fixes.