Simple JS fix