minor bug fixes that I forgot about