Bug Fixes

bug fixes