just trying to fix the tests