small tweak in testing code