Training sample transpose fix