Fix widget JS