Update the tests to match the new plugin structure.