fixing test for new schema