fix the sample also