Add null check for $host.UI and its properties to avoid race condition (#2383)

Add null check for $host.UI and its properties to avoid race condition (#2383) #384