Fix theme JS