fixed tests that were failing