fix 2 broken tests