Enable Update Management for Azure VMs with terraform

Hi,

I have an Azure automation account linked to a log analytics workspace with Update management enabled.
I am looking for a solution to onboard new VMs to the update management.
Is there a solution with terraform?

Thx