Bug Fixes

Bug fixes