Adjust testing to use new prove api