minor bug fix in test