Fix #18
Update comment