Fixing test to match changes.
Fixing test to match changes.