Cleaned test code.