fixed regression from last commit