bug fixes ....