Fixes regression in unit test.