Unit test fix.