Minor js fix