Fix the tests in the plugin module