Little edit of fix for #36