Quick fix for the second unit test.