Fixed outdated sample code