Returning new provision states with wait

Using wait=True means commands will complete before we return, so we can
return the final state.