minor test code update