Small js fix