bug fixes

pain