Fixing part of this test. Not working yet