bugfix: train - fix occasional mem leak in preview