little bug fix in testcode