Update arena plugin for tests