Trying to fix JS