Updated a test for datamodel changes.