small fix after refacto