Bug Fixes

Bug fixes.