Update test for the new spec