fix broken unit tests