Fix broken unit test from an earlier commit