Adjust timing test code