fix bug in test example