Fixed little js bug