Adds the lines that fix the issue