Small fix for #224