minor change to the code, fixing the .js error