Updated test code.