minor fix from last commit