fixes broken unit tests