Fixes broken unit tests.