minor bug fix from last commit