fixing the broken tests