Update test with new schema