Fix wrong sample code