Update test to use plugin