small fix following the previous commit