fixed tests

quick fix as there will be a new release pretty soon

closed #68 #72