fixing broken tests