fixed tests

quick fix as there will be a new release pretty soon

closed #68 #72
Your headers are killing me! Thanks