add 2 tests back