small fix from last commit