small change in testing code