bug fix tweaks