fixed the testing classes