(wip)[variational] Fix initialization