unbreak unit tests