Fixing custom JS