fixed tests which used depracated class