Bug fixes

Bug Fixes