Small change to the testing class.