Fixed vm creation for new dataset code.