Fixed a bug in sample code!