fixed the tests to actually work