Bug fixes

Bug fixes