A small fix for the previous commit