Bug fixes

