Fixes broken unit test.