tests bug fix