Update test to confirm the partial works