Fix heredoc issue