Fixes broken unit testing.