Fix JS references following updates