Fix/trainvm 10

Dunno if :metal: or :melon: 

Comments welcome.