fixes some broken tests