bug fixes

bug fixes