A small fix for the previous commit

For #1186