A small fix for previous commit