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.