Small suggestion to fix issue.

Hello,
Added to todo list. :slight_smile: