Modified test code.