Fixed code on sample