DisplayTest fixed after class change