bug fix in test