Fixing JS tests