small fix after last commit