The test cases need to be updated to the new api (#3)
The test cases need to be updated to the new api (#3)