Update tests to support subscription update