Add two more tests