Small fix for #170