[WIP] train
bug fix