Fixing unit test for customPlugin sample