Work on fixing test.