Fixing part of the test.