fix two more tests