fix the broken tests