Change test files to match plugin