Attempting to fix JS