bug fixes ...