fix tests, @lefnire not the best solution but it works