small fix to the previous commit