adjust test code