bug fixes....