hopefully fixed the tests