corrected broken tests