I'll update this test to correctly test the assertion with the new implementation details