Fix inconsistency on 'env.ui' vs. 'env[:vm].ui'

Using the example caused an error on a multiple vm environment.

As `env[:vm]` is used elsewhere I suggest it here as well.