Corrected test code.