fixed tests that broke