Update test for JSON changes