- fix broken tests