Update tests for the new API