Fix the test. todo: fix all the other tests.