fix remove training