whoops, fixed the tests