Test class modified