Fix class name
Add Avh/Di test