Fixed to work with core VE dialog changes

Change-Id: I429ad891de249a88116ad5358372cfc6a479afcb