Another small fix for #11