Fix test break due to module not included in expected classes.