Fix minor js bug