Misc. minor bug fixes