Custom.js fix