trainer: some logic fixes
Mutation