Fix JS in example