Fixing custom js bug.