Update use case to new API