Here is the fix from #5 comment.