updating tests with new schema