fix untrain