Fixing js script