bug fixes --QA