Fix for new Training_data class.