Fixes unit tests broken by previous commit.