fixed: dont train on root