TEST: fix broken tests