bug fixes

fix