Fix sample code for site testing