Fixing up test.