Fix JS a little