fixed broken tests.