bug fixes..