Fix test classes.