ChatGPT - Code issue fix