Updating test to make pig unit work.