Fix crash in unit tests.