fixes broken unit test