fixed the testing framework