fix broken tests in tests directory