fix to broken tests