fixed tests

quick fix as there will be a new release pretty soon