corrected the last two test classes