bug fix for test