Fixes old bug in unit tests.