quick fix on JS issue