Fix broken build due to unit test inconsistency following previous commit.