Update test case to match new example