Fix the unit test failure

Looks like I was testing the wrong intermediate behavior.