Fix failed test due to updated schema